Python 2d网格：将值分配给网格中的坐标

Question

我只是一个初学Python用户，所以请耐心等待，

我使用np.meshgrid创建了一个2d XY网格，并且想知道如何为网格中的某些点赋值。应用程序是将质量设置为网格中的某些点以模拟点质量。我现在的代码如下：

 TextView tv=findViewById(R.id.day_field)

非常感谢！

Answer 1

由于您是初学Python用户，我强烈建议您花点时间查看您正在使用的软件包的文档。这个example from matplotlib's documentation会回答你的问题吗？

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
ax = fig.gca(projection='3d')
X = np.arange(-5, 5, 0.25)
Y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm,
                       linewidth=0, antialiased=False)
ax.set_zlim(-1.01, 1.01)

ax.zaxis.set_major_locator(LinearLocator(10))
ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))

fig.colorbar(surf, shrink=0.5, aspect=5)

plt.show()

或者，您也可以查看meshgrid函数的numpy documentation，它提供了一个类似的示例：

x = np.arange(-5, 5, 0.1)
y = np.arange(-5, 5, 0.1)
xx, yy = meshgrid(x, y, sparse=True)
z = np.sin(xx**2 + yy**2) / (xx**2 + yy**2)
h = plt.contourf(x,y,z)

修改

如果您想手动设置Z的值，您可以执行以下操作

fig = plt.figure()
ax = fig.gca(projection='3d')
x = np.arange(11)
y = np.arange(11)
X, Y = np.meshgrid(x,y)
Z = np.zeros_like(X)
Z[5][5] = 5
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1)
plt.show()