numpy中的靈活n維索引

Question

我需要通過使用索引矩陣iM從n維矩陣中獲取一些元素。 但是這個n是未知的。

是否有可能獲得一種靈活的方式來做類似的事情：

if n == 1:
    value = M[iM[:]] # orM[iM[:, 0]]
elif n == 2:
    value = M[iM[:, 0], iM[:, 1]]
elif n == 3:
    value = M[iM[:, 0], iM[:, 1], iM[:, 2]]

更新：這可以像這樣完成：

value = M[[(iM[:, i]) for i in range(0, n)]]

Answer 1

您可以使用顯式list （例如tuple ）構造函數生成單個索引列表，因為這將自動將元素放入不同的列表中：

M[list(iM.T)] # or e.g. M[tuple(iM.T)]

由於列表的創建方式（逐行）和索引矩陣的布局（您希望逐列獲取索引），索引矩陣iM需要在其之前進行轉置（ iM.T ）變成一個列表。