从numpy矩阵中提取一行

Question

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

print train_set_data_vstacked_normalized.shape

(219970,400)

输入数据目前看起来像这样，我必须将每一行馈送到我的神经网络。

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

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

我试过了

print train_set_data_vstacked_normalized[:,0].shape
(219970,)

print train_set_data_vstacked_normalized[0,:].shape
(400,)

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.
   # ...

PS：作为旁注，我想提醒你，在我的机器中，内存中有~220K x 400阵列大约需要670 Mb。考虑将其作为HDF5文件。