python ndarray在特定索引處執行A [：，...，l，..：]

Question

假設我有一個變量但已知形狀的np.ndarray A。我想執行操作A[:, ... , :, l , :, ... : ]，其中l是一個已知的整數，出現在[...]的已知索引i處。

可能有一個相當簡單的解決方案，但我已經搜索了互聯網，找不到任何東西。

Answer 1

您可以在[]增加尺寸的arbitary數：

[(slice(None),) * i + (l,)] 

這是i:,然後l等同。例如：以

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

將相當於：

A[:, :, 3]