text_plain_output_1.png
import numpy as np

train = np.loadtxt('../input/train.csv', delimiter=',', skiprows=1)

train_label = train[:, 0]
train_img = np.resize(train[:, 1:], (train.shape[0], 28, 28))

train.shape

data_for_svd = train[:, 1:]
data_for_svd.shape