在Numpy中,meshgrid
函數用于生成網格點坐標矩陣。它接受一維的數組作為輸入,并返回兩個二維數組,這兩個數組分別對應輸入數組的行和列。
下面是meshgrid
函數的基本用法示例:
import numpy as np
x = np.array([1, 2, 3])
y = np.array([4, 5, 6])
x_grid, y_grid = np.meshgrid(x, y)
print(x_grid)
print(y_grid)
輸出結果為:
[[1 2 3]
[1 2 3]
[1 2 3]]
[[4 4 4]
[5 5 5]
[6 6 6]]
這里x_grid
是一個3x3的矩陣,其中每行的元素都是輸入數組x
的元素。y_grid
也是一個3x3的矩陣,其中每列的元素都是輸入數組y
的元素。這樣就生成了一個網格狀的坐標系。
meshgrid
函數還可以接受多個一維數組作為輸入,并生成多維網格矩陣。例如:
x = np.array([1, 2, 3])
y = np.array([4, 5, 6])
z = np.array([7, 8, 9])
x_grid, y_grid, z_grid = np.meshgrid(x, y, z)
print(x_grid)
print(y_grid)
print(z_grid)
輸出結果為一個3x3x3的三維矩陣,分別對應輸入數組x
,y
,z
的值在三維空間中的網格坐標。