是否有一个 numpy 函数来执行坐标转换？

Question

我正在 ipython 笔记本中开发一些关于数值方法的讲义，我需要包含一些坐标变换函数。我已经编写了这些函数（如下），但细节并不是主题的基础，如果我可以调用预先编写的函数，它将简化讲座。

我看到scipy图像旋转功能，但我只需要坐标平移和旋转。即

# transform from global to panel coordinates
def transformXY( self, x, y ):
    xp = x-self.xc
    yp = y-self.yc
    xpp = xp*self.sx+yp*self.sy
    ypp = yp*self.sx-xp*self.sy
    return [ xpp, ypp ]

# rotate velocity back to global coordinates
def rotateUV( self, u, v):
    up = u*self.sx-v*self.sy
    vp = v*self.sx+u*self.sy
    return [ up, vp ]  

其中，sx 和 sy 是单位切向量分量。

Answer 1

从之前的评论复制和扩展：

您还可以将自己的函数移动到单独的 .py 文件中，然后从那里导入它们。这样他们的内部就隐藏在笔记本中了。 如果将它们放在名为例如的文件/模块中transformation.py 并将该文件放在您的笔记本文件旁边，然后您可以使用 from transformation import * 导入您的函数

正如现在已删除的答案中所建议的那样，我建议使用矩阵进行任何转换，尤其是旋转。在我看来，这比元素方面的修改要清晰得多。

编辑： Afaik 在 numpy 中没有预定义的坐标变换函数。在 sympy 中有something，但我不确定它有多大用处，对于简单的转换来说它有点接缝。 对于通过矩阵进行的转换，谷歌找到了我这个module，这似乎很全面。