在python中填充数组答案

【问题标题】：Filling an array in python在python中填充数组
【发布时间】：2019-02-23 07:03:59
【问题描述】：

我想用一些数字填充一个数组，输出应该是这样的：

[[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 循环填充它，但我无法完全填充它。

如何在不手动填充的情况下使用 python 在 numpy 中进行操作？

【问题讨论】：

stackoverflow.com/a/44409182/7957705 可能就是您要找的东西
@FatihKılıç 我不认为它是，在他的情况下，输出将是[[a1 b1] [a2 b2] ... [a5 b5]]。就我而言，我想将每个 x 与每个 y 配对。另外，我的数据实际上远大于 2 x 和 2 y
@Amri Rasyidi 您的输出中有错字吗？第一列是 x1,x1,x2,x2 还是应该是 x1,x2,x1,x2 ？
@AndyK 不，这是我想要的输出。将每个 x 与每个具有 z 值的 y 配对

标签： python arrays numpy

【解决方案1】：

对于纯 Python 解决方案，适用于任何列表大小。

x = [11,12,13,14,15]
y = [21,23,24]
z = [31,32,33,34,35,36,37]

这个冗长的代码：

tmp = []
for i in range(max(len(x), len(y), len(z))):
    for yy in y:
        tmp.append([x[i%len(x)], yy])

for i in range(len(tmp)):
    tmp[i].append((z[i%len(z)]))

# [[11, 21, 31],
#  [11, 23, 32],
#  [11, 24, 33],
#  [12, 21, 34],
#  [12, 23, 35],
#  [12, 24, 36],
#  [13, 21, 37],
#  [13, 23, 31],
#  [13, 24, 32],
#  [14, 21, 33],
#  [14, 23, 34],
#  [14, 24, 35],
#  [15, 21, 36],
#  [15, 23, 37],
#  [15, 24, 31],
#  [11, 21, 32],
#  [11, 23, 33],
#  [11, 24, 34],
#  [12, 21, 35],
#  [12, 23, 36],
#  [12, 24, 37]]

【讨论】：

很抱歉，我已经编辑了我的问题。它是否适用于不同大小的 x 和 y 列表？
@AmriRasyidi 我编辑了，只留下纯 Python 选项适用于不同大小的任何列表。

【解决方案2】：

分别在 x 和 y 上使用 np.repeat 和 np.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]])

【讨论】：

我检查了dtype，它返回'U10'，我还能有数据的数字格式吗？
@Amri Rasyidi 你的数据是什么样的？ x,y 和 z 的值是否属于同一类型？它们的值是字符串吗？它仍然应该工作。
我找到了解决该 dtype 问题的方法。我将其转换为数组，然后使用 astype 更改数据类型。谢谢！