text_html_output_1.png text_plain_output_1.png
import pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)

traindf = pd.read_csv('../input/train.csv')
testdf = pd.read_csv('../input/test.csv')
totalRows = traindf[' y'].max()
totalCols = traindf['x'].max()
print('room size ', totalRows, ' ', totalCols)
trainData = traindf.iloc[:, 2:]
trainlabels = traindf.iloc[:, 0:2]
testData = testdf.iloc[:, 2:]
testlabels = testdf.iloc[:, 0:2]
dimOfdata = trainData.shape[1]
print('number of  features: ', dimOfdata)
numberOfSamples = len(trainlabels)
print('total samples: ', numberOfSamples)