ndarrayの要素の指定について

Question

プログラミング初心者です。

Pythonでnumpyのndarrayを使い計算を行っているのですが、２次元配列の要素の指定について思うように指定できず困っています。

a = np.arange(0, 100)
arr = np.reshape(a,(10, 10))
i = np.arange(0, 10, 2)
j = np.arange(1, 10, 2)
print(arr[i,j])

とした際に、出力が

[[ 1  3  5  7  9]
 [21 23 25 27 29]
 [41 43 45 47 49]
 [61 63 65 67 69]
 [81 83 85 87 89]]

となることを期待したのですが、実際は

[ 1 23 45 67 89]

となってしまいます。

print(arr[0:10:2, 1:10:2])

とすれば希望通りの出力になるのですが、何度も使うため読みやすさを考えできれば避けたいのですが方法はありますでしょうか？

初歩的な質問かもしれませんがアドバイスを頂ければ幸いです。

Answer 1

numpy.ix_にて要素の一部を抽出できます。
参考：[Python]Numpyの参照、抽出、結合

import numpy as np
a = np.arange(0, 100)
arr = np.reshape(a,(10, 10))
i = np.arange(0, 10, 2)
j = np.arange(1, 10, 2)
print(arr[np.ix_(i,j)])
"""
[[ 1  3  5  7  9]
 [21 23 25 27 29]
 [41 43 45 47 49]
 [61 63 65 67 69]
 [81 83 85 87 89]]
"""

Comments

[R_SSK]

ありがとうございます！参考先は閲覧していたのですが理解できていませんでした汗
またよろしければご教授お願いします。