Python中的網格工具（meshgrid mgrid ogrid ndgrid）

Question

我正在尋找類似網狀網格功能的清晰對比。 不幸的是我找不到它！

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

• mgrid

• ogrid

• meshgrid

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所有組合。 考慮一下，在廣播時，對於numpy數組， meshgrid有點多余。 這意味着你可以做

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實際上是幫助此類生成的整個類。

Answer 2

np.mgrid和np.meshgrid()做相同的事情，但是交換了第一根和第二根軸：

# 3D
d1, d2, d3 = np.mgrid[0:10, 0:10, 0:10]
d11, d22, d33 = np.meshgrid(np.arange(10),np.arange(10),np.arange(10))
np.array_equal(d1,d11)

產生False 。 只需交換前兩個維度：

d11 = np.transpose(d11,[1,0,2])
np.array_equal(d1,d11)

產生True 。