如何在不使用Python循环的情况下使用每一行中的指定列从矩阵创建向量？

Question

假设我有一个形状为（N，d）的矩阵和一个大小为N的向量，该向量表示矩阵中的哪一列与给定行相关。如何返回矩阵和相关列中的值所给定的大小为N的向量？

例如：

M = [[ 2, 4, 1, 8],
    [3, 5, 7, 1],
    [2, 5, 3, 9],
    [1, 2, 3, 4]]
V = [2, 1, 0, 1]

我尝试了类似的东西：

M[:,V]

但这返回的矩阵是NXN

有没有一种简单的格式化方法，它不涉及编写for循环，因此我可以获得以下向量：

V' = [1,5,2,2]

Answer 1

使用np.arange(len(V))用于索引的行数和V列的：

In [110]: M = [[ 2, 4, 1, 8],
   .....:     [3, 5, 7, 1],
   .....:     [2, 5, 3, 9],
   .....:     [1, 2, 3, 4]]

In [111]: V = [2, 1, 0, 1]

In [112]:

In [112]: M = np.array(M)

In [113]: M[np.arange(len(V)),V]
Out[113]: array([1, 5, 2, 2])