首页 > 解决方案 > 附加数组的numpy数组

问题描述

我试图将一个数组附加到另​​一个数组,但它附加它们就好像它只是一个数组一样。我想要的是每个数组都附加在自己的索引上,(不必使用列表,我想使用 np 数组)即

temp = np.array([])
for i in my_items
   m = get_item_ids(i.color)  #returns an array as [1,4,20,5,3]  (always same number of items but diff ids
   temp = np.append(temp, m, axis=0)

在第二次迭代中,假设我得到 [5,4,15,3,10]

然后我想有温度 array([1,4,20,5,3][5,4,15,3,10]) 但相反我一直在[1,4,20,5,3,5,4,15,3,10]

我是 python 新手,但我确信可能有一种方法可以在不使用列表的情况下以这种方式与 numpy 连接?

标签: pythonnumpy

解决方案


您必须重塑 m 才能拥有二维

m.reshape(-1, 1)

从而增加了第二个维度。然后你可以沿着axis = 1连接。

np.concatenate(temp, m, axis=1)

推荐阅读