Numpy dstack вызывает ошибку памяти

Question

У меня есть два массива фигуры:

x.shape # (50000, 784)
y.shape # (50000,)

Однако, когда я использую column_stack, я получаю MemoryError

X_y = np.column_stack([X, y])
# ... MemoryError.

Мой желаемый результат - добавить y в качестве другого столбцак X

X_y.shape # (50000, 785)

Я также пытался

X_y = list(zip(X, Y)) 
random.shuffle(X_y)
X, Y = zip(*X_y)
X, Y = np.array(X), np.array(Y)

И это также вызывает MemoryError в последней строке

Способ, которым я получаю эту ошибкувыглядит следующим образом:

TrainX = np.loadtxt("TrainDigitX.csv.gz", delimiter=",")
TrainY = np.loadtxt("TrainDigitY.csv.gz", delimiter=",")

NN = Neural_Network((784, 30, 10))
mini_batch_size = 20
epochs = 30

for epoch in range(epochs):
    TrainX = TrainX.T.tolist()
    TrainX.append(TrainY)
    X_match_y = np.array(TrainX).T
    np.random.shuffle(X_match_y)
#... continue to run NN