numpy 层压两列阵列

bxjv4tth 于 2022-12-18 发布在其他

关注(0)|答案(1)|浏览(114)

假设我有两个numpy（N，1）列数组，我知道它们的长度相等：

>>> a
array([[0.],
       [2.],
       [4.],
       [6.]])
>>> b
array([[0.],
       [1.],
       [2.],
       [3.]])

我想把它们并排“层压”成一个（N，2）数组。

>>> np.array((a, b)).reshape(2, len(a)).transpose()
array([[0., 0.],
       [2., 1.],
       [4., 2.],
       [6., 3.]])

......但有没有更简单、更直接的方法来实现这一点？

1条答案

您可以使用np.hstack（docs）将它们水平堆叠

>>> np.hstack((a, b))
array([[0., 0.],
       [2., 1.],
       [4., 2.],
       [6., 3.]]