Numpyのndarrayの列の組み合わせを表示させたい

Question

いま、A

array([[ -2.5,  -1.5,  -0.5,   0.5,   1.5,   2.5,   3.5,   4.5,   5.5,
          6.5],
       [  2. ,  -2. ,   4. ,  -4. ,   6. ,  -6. ,   8. ,  -8. ,  10. ,
        -10. ],
       [-30. , -20. , -10. ,   5. ,  40. ,  30. ,  20. ,  10. ,   5. ,
          5. ],
       [  1. ,   1. ,   1. ,   1. ,   1. ,   1. ,   1. ,   1. ,   1. ,
          1. ]])

というndarrayがあり、その列ごとの組み合わせを表示させたいと考えております。
例えば、縦の[[-2.5], [2.], [-30], [1.]], [[-0.5], [-2], [-20], [1]]という風な組み合わせです。
このAは10列あり、その中から４列選ぶ組み合わせを表示させたいので、210通りの配列を表示させたいと考えております。
itertoolsを使って行ごとの組み合わせの表示はできたのですが、列ごとの組み合わせの表示はどのように行うのでしょうか？

回答よろしくお願いいたします。

Comments

[tiitoi]

「列ごとの組み合わせ」とは具体的にはどのような出力を想定されてますか？

[raithend]

例えば、縦の[[-2.5], [2.], [-30], [1.]], [[-0.5], [-2], [-20], [1]]という風な組み合わせです。
このAは10列あり、その中から４列選ぶ組み合わせを表示させたいので、210通りの配列を表示させたいと考えております。

Answer 1

手順

• 1. itertools.combination(行列の列数, 抽出する列数) で列の組み合わせを生成する。
• 1. ndarray[:, 抽出する列のインデックスのリスト] で一部を取得する。

例

a = np.array([[1, 2, 3],
              [4, 5, 6]])
print(a[:, [0, 2]])  # 0, 2 列だけ抽出
# [[1 3]
#  [4 6]]

サンプルコード

import itertools

import numpy as np

arr = np.array(
    [
        [-2.5, -1.5, -0.5, 0.5, 1.5, 2.5, 3.5, 4.5, 5.5, 6.5],
        [2.0, -2.0, 4.0, -4.0, 6.0, -6.0, 8.0, -8.0, 10.0, -10.0],
        [-30.0, -20.0, -10.0, 5.0, 40.0, 30.0, 20.0, 10.0, 5.0, 5.0],
        [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0],
    ]
)

for cols in itertools.combinations(range(arr.shape[1]), 4):
    print(arr[:, cols])

Comments

[raithend]

できました！
ありがとうございます！