用两个1维数组制作2维numpy数组

时间:2018-12-13 10:53:59

标签: python arrays numpy multidimensional-array

我想转换2个numpy数组,例如这些数组:

a = [[1, 2, 3]]
b = [[100, 200, 300]]

到如下数组。

[[1, 100], [1, 200], [1, 300], [2, 100], [2, 200], [3, 300], [3, 100], [3, 200], [3, 300]]   

在NumPy中有可能吗?

谢谢。

(已编辑以阐明此问题的要点。) 我正在尝试找到一种解决方法。

1 个答案:

答案 0 :(得分:1)

这是meshgridstack的工作:

a = np.array([ [1, 2, 3] ])
b = np.array([ [100, 200, 300] ])

print(np.stack(np.meshgrid(a, b)).T.reshape(-1,2))

第一个在网格上创建一个元组坐标,第二个将它们堆叠。然后,您只需要转置和展平即可。