擅长:python、mysql、java
<p>您的函数很小,不需要参数,所以您真的需要将绘图包装在函数中吗?怎么办:</p>
<pre><code>import numpy as np
import matplotlib.pyplot as plt
fig, ax = plt.subplots(1, 1)
x, y = np.mgrid[0:5,0:5]
z = np.sin(x**2+y**2)
mesh = ax.pcolormesh(x, y ,z)
fig.colorbar(mesh)
plt.show()
</code></pre>
<p><img src="https://i.stack.imgur.com/8TneQ.png" alt="enter image description here"/></p>