使用numpy将两个数组垂直合并为元组数组

Question

我有两个numpy数组：

x = np.array([-1, 0, 1, 2])
y = np.array([-2, -1, 0, 1])

有没有一种方法可以将这些数组像元组一样合并在一起：

array = [(-1, -2), (0, -1), (1, 0), (2, 1)]

- Steven

Answer 1

In [469]: x = np.array([-1, 0, 1, 2])
In [470]: y = np.array([-2, -1, 0, 1])

将它们组合成二维数组：

In [471]: np.array((x,y))
Out[471]: 
array([[-1,  0,  1,  2],
       [-2, -1,  0,  1]])

将该数组转置：

In [472]: np.array((x,y)).T
Out[472]: 
array([[-1, -2],
       [ 0, -1],
       [ 1,  0],
       [ 2,  1]])

或者使用标准的Python zip - 这会将数组视为列表

In [474]: zip(x,y)   # list(zip in py3
Out[474]: [(-1, -2), (0, -1), (1, 0), (2, 1)]

- hpaulj

---

这些结果不是元组数据类型。 - john k

@johnktejik，“Out[474]”是一个元组列表。“Out[472]”不是，但对于许多目的来说它同样好用——包括OP的目的。在创建结构化数组时，元组列表和列表列表之间的区别很重要，但这是一个例外。 - hpaulj

对于Python 3，zip函数返回一个迭代器而不是列表。可以使用“list(zip(x,y))”将其转换为列表。 - Mathi

Answer 2

可以通过np.c_[]在第二个维度上进行连接。

x = np.array([-1, 0, 1, 2])
y = np.array([-2, -1, 0, 1])

xy = np.c_[x, y]

# array([[-1, -2],
#        [ 0, -1],
#        [ 1,  0],
#        [ 2,  1]])

如果你需要一个元组数组，那么有记录数组：

xy = np.rec.fromarrays([x, y])
# rec.array([(-1, -2), ( 0, -1), ( 1,  0), ( 2,  1)], dtype=[('f0', '<i4'), ('f1', '<i4')])

# convert into a list
xy.tolist()   # [(-1, -2), (0, -1), (1, 0), (2, 1)]

- cottontail