image_output_1.png
import pandas as pd

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

train_images = train.ix[:, 1:].values.astype('float32')
train_labels = train.ix[:, 0].values.astype('int32')

train_labels.shape