我想转换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中有可能吗?
(已编辑以阐明此问题的要点。) 我正在尝试找到一种解决方法。
答案 0 :(得分:1)
这是meshgrid
和stack
的工作:
a = np.array([ [1, 2, 3] ])
b = np.array([ [100, 200, 300] ])
print(np.stack(np.meshgrid(a, b)).T.reshape(-1,2))
第一个在网格上创建一个元组坐标,第二个将它们堆叠。然后,您只需要转置和展平即可。