python - 创建一个基于空的numpy数组并附加现有数组

Question

我想初始化空的numpy（E）数组，然后将现有数组（D）中的选定行附加到空数组（E）。我无法初始化与 D 的维度匹配的空数组。我该如何解决这个问题。

E = np.array([ [ ] ])

D = np.array([ [20, 23 ], [19, 23],[19,22] ]) 

for i in np.where(index):  ## assume index value is ([0],[2])

    E = np.append( E , D[i] , axis = 0)

我试图达到以下结果：

E = ([ [20, 23 ], [19,22] ]) 

错误：除连接轴外的所有输入数组维度必须完全匹配

标签： pythonarrayslistnumpy

Answer 1

为什么要初始化一个空数组？您可以D根据索引进行切片：

index = [0,2]

E = D[index]

>>> E
array([[20, 23],
       [19, 22]])