python - Python 中的网格网格函数 (meshgrid mgrid ogrid ndgrid)

Question

我正在寻找类似网格的功能的清晰比较。可惜没找到!

Numpy http://docs.scipy.org/doc/numpy/reference/提供

• mgrid

• ogrid

• 网格

Scitools http://hplgit.github.io/scitools/doc/api/html/index.html提供

• ndgrid

• boxgrid

理想情况下，总结所有这些的表格是完美的!

Answer 1

numpy.meshgrid 仿照 Matlab 的 meshgrid 命令。它用于向量化两个变量的函数，以便您可以编写

x = numpy.array([1, 2, 3])
y = numpy.array([10, 20, 30]) 
XX, YY = numpy.meshgrid(x, y)
ZZ = XX + YY

ZZ => array([[11, 12, 13],
             [21, 22, 23],
             [31, 32, 33]])

所以ZZ包含了放入函数中的所有x和y的组合。仔细想想，meshgrid 对于 numpy 数组来说有点多余，因为它们是广播的。这意味着你可以做到

XX, YY = numpy.atleast_2d(x, y)
YY = YY.T # transpose to allow broadcasting
ZZ = XX + YY

并得到相同的结果。

mgrid 和 ogrid 是使用索引符号的辅助类，因此您可以在其中创建 XX 和 YY直接使用前面的示例，而不必使用 linspace 之类的东西。生成输出的顺序是相反的。

YY, XX = numpy.mgrid[10:40:10, 1:4]
ZZ = XX + YY # These are equivalent to the output of meshgrid

YY, XX = numpy.ogrid[10:40:10, 1:4]
ZZ = XX + YY # These are equivalent to the atleast_2d example

我不熟悉 scitools 的东西，但是 ndgrid 似乎等同于 meshgrid，而 BoxGrid 实际上是一个完整的类来帮助这一代人。