将特定索引从3D numpy数组返回到1D

2024-04-26 05:18:58 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个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\u new\u array=np.split(x\u train,990,axis=0) 但是我得到了一个列表(990),每个记录都有一个类型数组(1,19,13)

它不能解决我的问题

除了这个解决方案之外,我还在堆栈中检查了一些解决方案 从这里来的,但有些东西我不明白,如果你能帮忙我真的很感激


Tags: 数据类型列表newmynp记录train