numpyで各次元に対する操作を行う方法はありますか？

Question

numpyで各次元に対して操作を行いたいです。
例えば、

#各次元の初めの要素に対して操作を行う
array = np.zeros((5,5,5,5,5))
for i in range(len(array.shape)):
    print(array[...,0,i*(:)])

こんな感じです。evalを使えば早そうですがevalには抵抗があり、このような操作を実現する方法があれば、ご教授願いたいです。

Answer 1

take()でaxisを指定すればできます。
https://numpy.org/doc/stable/reference/generated/numpy.ndarray.take.html
https://numpy.org/doc/stable/reference/generated/numpy.take.html

for i in range(len(array.shape) - 1, -1, -1):
    print(array.take(0, axis=i))

Comments

[37458]

できました。ありがとうございます：）

Answer 2

特殊メソッド__getitem__を使って、以下のようにします。

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

 [[12 13 14 15]
  [16 17 18 19]
  [20 21 22 23]]]
>>>
>>> for i in range(len(array.shape)):
...     print(array.__getitem__(tuple([Ellipsis, 0] + [slice(None, None, None)]*i)))
...
[[ 0  4  8]
 [12 16 20]]
[[ 0  1  2  3]
 [12 13 14 15]]
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]

Comments

[37458]

ありがとうございます。そのメソッドは初めて知りました。[]はそのメソッドを呼び出してるんですね！