pythonのリスト操作について

Question

Aを5000行×7列のNumpy配列とします。

A.shape=(5000,7)

Aに対して以下のような処理をした場合の挙動を教えて下さい。また、どういった意味があるのでしょうか。
リストの要素指定と思うのですが、二重の括弧で囲むパターンを初めて見たため、お分かりになる方がおられれば回答をお願いします。

for i in range(len(A)):
B=A[ [ i ] ]

Answer 1

以下のような挙動になります。

>>> import numpy as np
>>> a = np.arange(9).reshape(3, 3)
>>> a[[0]]
array([[0, 1, 2]])
>>> a[[0, 2]]
array([[0, 1, 2],
       [6, 7, 8]])
>>> a[[1, 0]]
array([[3, 4, 5],
       [0, 1, 2]])

配列のindexingに直接カンマを使っていない（外側の角括弧の直下でtupleにしていない）ので、aixs=0への指定だとわかります。iterableを書いた場合、その要素の行を取り出せということです。

Indexing — NumPy v1.17 Manual

Comments

[akawase]

丁寧なご回答ありがとうございます。
配列の要素指定をリストによって行う、所謂ファンシーインデックスということだと思います。
この方法は複数の行を抽出するには便利ですが、今回は１つの行のみです。
それであれば、a[i]のような記法で足りるように思うのですが...
回答を見たところ、1次元配列でなく2次元配列になっています。この辺りが関係しているのでしょうか。

[hayataka2049]

そうですね、結果のndimが変わりますね。

[akawase]

ありがとうございます、大変助かりました。

Answer 2

二重の括弧で囲むパターンを初めて見たため、

二重の括弧という特別な記法があるわけじゃないです。下記と同じです。

for  i  in  range(len(A)):
    q = [i]
    B = A[q]

Comments

[akawase]

ありがとうございます、参考にさせて頂きます。