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

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

使用

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])