將3D numpy數組中的特定索引返回到1D

Question

我有一個3D數組，我需要從特定的軸恢復對應於一個特定索引的數據。

x_train.shape
Out[101]: (990, 19, 3)

x_train[:,0,:]
Out[117]: 
array([[0.08581368, 0.09640129, 0.09044931],
       [0.09701243, 0.09834351, 0.09823458],
       [0.09433366, 0.09251685, 0.09408623],
       ...,
       [0.88172483, 0.9100043 , 0.89555236],
       [0.90901481, 0.9104525 , 0.90519397],
       [0.89063546, 0.90311315, 0.90350811]])

我需要從這個數組中只得到索引0

0.08581368,
0.09701243,
0.09433366,
...,
0.90901481,
0.89063546

我嘗試用my_new_array = np.split（x_train，990，axis = 0）拆分數組，但我得到一個列表（990），每個記錄我都有一個類型數組（1,19,13）

它不能解決我的問題。

我已經在堆棧中檢查了一些解決方案，除了這個解決方案來自這里，但有些東西我沒有得到，如果你能提供幫助我真的很感激。

Answer 1

第一個索引是x_train[:, 0, :][0] ，它可以通過直接訪問

x_train[0, 0, :]

這是第一行 - 但這不是你想要的。

您的示例看起來像第一列 ，可以通過以下方式訪問

x_train[:, 0, 0]