从numpy矩阵中提取一行

Question

我目前正在测试一个NN实现，其中列车数据存储在numpy矩阵中。在

print train_set_data_vstacked_normalized.shape

(219970,400)

当前的输入数据是这样的，我必须把每一行输入我的神经网络。。在

它接受形状的输入（无，400）。在

如何取出一行，这样我从矩阵中取出的数组有400个条目，或者400列和一行？在

我试过了

^{pr2}$

Answer 1

您需要一个简单的for循环来遍历数组的所有行。在

nrows = train_set_data_vstacked_normalized.shape
for i in range(nrows[0]):
    row = train_set_data_vstacked_normalized[i, :]

    # now change shape to (1, 400)
    resized_row = row[np.newaxis]

   # now, "resized_row" shape is (1, 400)
   # pass "resized_row" to NN input layer.
   # ...

附言：顺便提醒一下，在内存中使用~220K x 400数组在我的机器中大约需要670mb。考虑将此文件作为HDF5文件。在