配列の作り方を一般化したい．

Question

前提・実現したいこと

以下のソースコードで行っていることを一般化したい．

発生している問題・エラーメッセージ

以下のソースコードで出力される(5^3, 3)の配列を(5^n, n)の配列にしたいのですが，それを作る方法がわかりません．

該当のソースコード

import numpy as np
b = [1, 2, 3, 4, 5]
a = []
cala = [[]] * 3
cala1 = []
cala2 = []
cala3 = []
for j in range(len(b)):
    for k in range(len(b)):
        for l in range(len(b)):
            cala1.append(b[j])
            cala2.append(b[k])
            cala3.append(b[l])
a = (cala1, cala2, cala3)
choko = np.transpose(a)

試したこと

再帰関数を用いて，以下のようにやりました．objを初期化してしまってうまくいかないのはわかりますが，どう直したらいいのかよくわからないです...

import numpy as np
def re(n, a):
    obj = []
    if n <= 0:
        return a
    else:
        coo = []
        for i in range(len(a)):
            coo.append(a[i])
        obj.extend(coo)
        return re(n-1, obj)
n = 3
a = [1, 2, 3, 4, 5]
p = re(n, a,)
p = np.transpose(p)

補足情報（FW/ツールのバージョンなど）

Pythonのバージョンは3.6.4です．

Answer 1

itertoolsのproductを使うといいです。

from itertools import product
b = [1, 2, 3, 4, 5]
n = 3
choko = list(list(x) for x in product(b, repeat=n))

Comments

[退会済みユーザー]

ありがとうございます！
numpy形式の場合もよかったら教えていただきたいです．

[tachikoma]

np.array(choko)ではなく？