python-3.x - 如何组合两个不同类型的数组并将它们作为列表

Question

我有两个 numpy 数组如下

A = [1,2,3,1,2,3,1,2,3] #integers
B = ['xx','xx','xx','yy','yy','yy','zz','zz''zz'] #strings

我想组合并存储为一个列表，例如：

AB_list = [[1,'xx'],[2,'xx'],[3,'xx'],[1,'yy'],[2,'yy'],[3,'yy'],[1,'zz'],[2,'zz'],[3,'zz'],]

任何人都可以帮忙吗？

标签： python-3.xlistnumpy

Answer 1

像这样使用列表理解和 zip 迭代器的东西应该可以工作：

A = np.array([1,2,3,1,2,3,1,2,3]) #integers
B = np.array(['xx','xx','xx','yy','yy','yy','zz','zz','zz'])
[ [a,b] for a,b in zip(A,B) ]
Out[29]: 
[[1, 'xx'],
 [2, 'xx'],
 [3, 'xx'],
 [1, 'yy'],
 [2, 'yy'],
 [3, 'yy'],
 [1, 'zz'],
 [2, 'zz'],
 [3, 'zz']]