Numpy Row by Row Rotation

时间:2017-10-30 00:17:57

标签: python numpy rotation

我有一个500x2 numpy数组和一个2x2旋转数组。 我想使用旋转数组旋转数组的每一行。

我试过了

R.dot(A)

其中R是旋转数组,A是500x2数组。但是得到以下错误:

shapes (2,2) and (500,2) not aligned: 2 (dim 1) != 500 (dim 0)

有没有办法逐行执行此操作?

由于

1 个答案:

答案 0 :(得分:2)

对于矩阵乘法,您需要匹配内部维度(m nn k)。您试图将2乘2矩阵与500乘2矩阵相乘 - 您需要对第二个矩阵进行转置。尝试

R.dot(A.transpose())

结果的列应该是旋转的矢量。如果你需要它以行形式,你可以再次进行转置:

R.dot(A.transpose()).transpose()