如何使用每行中的特定列从矩阵创建向量而不在 Python 中循环？ [复制]答案

【问题标题】：How can I create a vector from a matrix using specified colums from each row without looping in Python? [duplicate]如何使用每行中的特定列从矩阵创建向量而不在 Python 中循环？ [复制]
【发布时间】：2017-02-03 04:29:03
【问题描述】：

假设我有一个形状为 (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]

【问题讨论】：

标签： python numpy matrix indexing

【解决方案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])

【讨论】：