python中 meshgrid函数的使用
时间: 2023-09-16 20:13:59 浏览: 103
在 Python 中,`numpy.meshgrid()` 函数用于生成网格点坐标矩阵。我们可以使用这个函数来生成网格点坐标矩阵,以方便进行二维函数的绘图和计算。
`numpy.meshgrid()` 函数的语法为:
```python
numpy.meshgrid(*xi, **kwargs)
```
其中,`*xi` 表示输入的一维数组,`**kwargs` 表示可选参数。
下面是一个简单的例子,演示如何使用 `numpy.meshgrid()` 函数:
```python
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(-5, 5, 0.1)
y = np.arange(-5, 5, 0.1)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))
plt.contourf(X, Y, Z)
plt.colorbar()
plt.show()
```
这个例子中,我们使用 `numpy.arange()` 函数生成了两个一维数组 `x` 和 `y`,然后使用 `numpy.meshgrid()` 函数生成了网格点坐标矩阵 `X` 和 `Y`,最后计算了二维函数 `sin(sqrt(x^2+y^2))` 的值,并用 `matplotlib` 库绘制了等高线图。
在这个例子中,`numpy.meshgrid()` 函数的作用就是将一维数组 `x` 和 `y` 转换成二维数组 `X` 和 `Y`,使得我们可以方便地计算二维函数的值。
阅读全文