Question

我有2个numpy数组x1和x2。使用python 3.4.3

x1 = np.array([2,4,4])
x2 = np.array([3,5,3])

我想要一个像这样的numpy数组：

[[2,3],[4,5],[4,3]]

我该怎么做？

Answer 1

您可以使用numpy.column_stack：

In [40]: x1 = np.array([2,4,4])

In [41]: x2 = np.array([3,5,3])

In [42]: np.column_stack((x1, x2))
Out[42]: 
array([[2, 3],
       [4, 5],
       [4, 3]])

Answer 2

是的。听起来像拉链功能：

import numpy as np 

x1 = np.array([2,4,4])
x2 = np.array([3,5,3])

print zip(x1, x2) # or [list(i) for i in zip(x1, x2)]

Answer 3

您可以zip这两个数组：

x3 = list(zip(x1,x2))

输出：

[(2, 3), (4, 5), (4, 3)]

以上代码会创建list tuples。如果您需要list lists，则可以使用list comprehension：

x3 = [list(i) for i in list(zip(x1,x2))]

输出：

[[2, 3], [4, 5], [4, 3]]

Python Numpy数组转换

3 个答案: