python ndarray在特定索引处执行A [:,...,l,..:]

时间:2017-04-08 00:08:28

标签: python numpy multidimensional-array

假设我有一个变量但已知形状的np.ndarray A。我想执行A[:, ... , :, l , :, ... : ]操作,其中l是一个已知整数,它出现在i中的已知索引[...]

可能有一个相当简单的解决方案,但我已经搜索了互联网并且找不到任何东西。

1 个答案:

答案 0 :(得分:2)

您可以在[]

中添加任意数量的维度
[(slice(None), ) * i + (l, )]

这相当于i :,,然后是l。例如:

i = 2
l = 3
A[(slice(None), ) * i + (l, )]

相当于:

A[:, :, 3]