我想用一些数字填充数组,输出应如下所示:

[[x1, y1, z1],
 [x1, y2, z2],
 [x2, y1, z3],
 [x2, y2, z4],
 [x3, y1, z5],
 [x3, y2, z6]]


我有数据x,y和z(列表),其中有3个值x,2个y值和6个z值。

我试图创建一个新数组,然后使用for或while循环填充它,但无法完全填充它。

如何在numpy中使用python而不手动填充它?

最佳答案

分别在x和y上使用np.repeatnp.tile以获得所需的列,然后将它们与z一起堆叠到一个数组中:

import numpy as np

x = [1, 2, 3]
y = [4, 5]
z = [6, 7, 8, 9, 10, 11]

a = np.array([np.repeat(x, 2), np.tile(y, 3), z]).T


结果:

array([[ 1,  4,  6],
       [ 1,  5,  7],
       [ 2,  4,  8],
       [ 2,  5,  9],
       [ 3,  4, 10],
       [ 3,  5, 11]])

08-26 18:12
查看更多