python - numpy マトリックスから 1 行を抽出する

Question

私は現在、列車データがnumpy行列に格納されているNN実装をテストしています。

print train_set_data_vstacked_normalized.shape

(219970,400)

現在、入力データは次のようになっています。各行をニューラル ネットワークにフィードする必要があります。

形状（なし、400）の入力を受け付けます。

行列から取り出す配列に 400 エントリ、または 400 列と 1 行が含まれるように、1 行を取り出すにはどうすればよいですか?

私はもう試した

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

220K x 400PS: 余談ですが、~配列をメモリに格納 すると、私のマシンでは約670 Mbかかることを思い出してください。これを HDF5 ファイルとして使用することを検討してください。