如何将两个矩阵向量乘以熊猫或numpy中矩阵的特定索引处的向量

时间:2018-06-28 08:17:22

标签: python pandas numpy matrix

假设我们有两个矩阵A和B分别具有P * N和N * Q的暗淡,那么仅在特定索引下从这两个矩阵相乘得到值的更有效方法是什么?例如,矩阵A中{p1,p3,p6,p1,...}的行索引向量与矩阵B中{q2,q5,q2,q3 ...}的列索引向量相乘,并返回标量值。

还有一个数据框,用于存储这些行索引和列索引的映射,例如,

row,column
p1,q2
p3,q5
p6,q2
p1,q3
...

就像前面提到的。

1 个答案:

答案 0 :(得分:0)

首先,我将从矩阵(2-D数组)中过滤出所需的行和列。然后,您可以使用简化后的矩阵执行矩阵乘法,并根据需要将结果向量分配给标量变量。

关于矩阵乘法,请看numpy.matmul

可以通过多种方式从矩阵中获取行和列。这个thread 已经使用numpy._ix广播为您的问题提供了一些答案。