任意の配列n個の中身をループを回して１つずつ全通りの組合せを表示する。

やったこと

任意の配列n個の中身をループを回して１つずつ全通りの組合せを表示する。

narray.py

# coding:utf-8
import itertools

dim_num = 3
b = [chr(i) for i in range(65,65+dim_num)]

for b in itertools.product(range(3),repeat=dim_num):
    print(b)

実行結果

(0, 0, 0)
(0, 0, 1)
(0, 0, 2)
(0, 1, 0)
(0, 1, 1)
(0, 1, 2)
(0, 2, 0)
(0, 2, 1)
(0, 2, 2)
(1, 0, 0)
(1, 0, 1)
(1, 0, 2)
(1, 1, 0)
(1, 1, 1)
(1, 1, 2)
(1, 2, 0)
(1, 2, 1)
(1, 2, 2)
(2, 0, 0)
(2, 0, 1)
(2, 0, 2)
(2, 1, 0)
(2, 1, 1)
(2, 1, 2)
(2, 2, 0)
(2, 2, 1)
(2, 2, 2)

こうとかもいける。

narray.py

# coding:utf-8
import itertools
import numpy as np

dim_num = 3
b = [chr(i) for i in range(65,65+dim_num)]
array = np.array(([[[3,3],[2,1]],[[2,1],[3,2]]]))

for b in itertools.product(range(2),repeat=dim_num):
    print(array[b])

実行結果

[[[3 3]
  [2 1]]

 [[2 1]
  [3 2]]]
3
3
2
1
2
1
3
2

参考文献

アルファベットのリストを作る in python
itertools — 効率的なループ実行のためのイテレータ生成関数