Заполнение массива в python

Question

Я хочу заполнить массив некоторым числом, вывод должен быть таким:

[[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, не заполняя его вручную?

Answer 1

Используя np.repeat и np.tile для x и y соответственно, чтобы получить нужные столбцы, а затем объединить их вместе с 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]])

Answer 2

Для чистого решения 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]]