假设我有一个变量但已知形状的np.ndarray A
。我想执行A[:, ... , :, l , :, ... : ]
操作,其中l
是一个已知整数,它出现在i
中的已知索引[...]
。
可能有一个相当简单的解决方案,但我已经搜索了互联网并且找不到任何东西。
答案 0 :(得分:2)
您可以在[]
:
[(slice(None), ) * i + (l, )]
这相当于i
:,
,然后是l
。例如:
i = 2
l = 3
A[(slice(None), ) * i + (l, )]
相当于:
A[:, :, 3]