text_plain_output_1.png
import numpy as np # linear algebra

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

train_data = pd.read_csv('../input/train.csv')
test_data = pd.read_csv('../input/test.csv')

x_train = train_data.drop(labels='label', axis=1)

print('number of training examples', x_train.shape[0])
print('number of test examples', test_data.shape[0])
print('number of pixels', x_train.shape[1])
print('number of rows and columns', int(np.sqrt(x_train.shape[1])))