要使用numpy或pandas生成带有Z值的2D网格(X,Y),你可以按照以下步骤操作:
import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)
Z = X**2 + Y**2
plt.contourf(X, Y, Z, levels=20, cmap='RdGy')
plt.colorbar()
plt.show()
虽然pandas不是专门用于数学计算的库,但你仍然可以使用它来生成2D网格并计算Z值。不过,通常情况下,numpy更适合这种任务。
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
df = pd.DataFrame({'X': X.ravel(), 'Y': Y.ravel(), 'Z': Z.ravel()})
X_grid, Y_grid = np.meshgrid(x, y)
Z_grid = df.pivot_table(index='Y', columns='X', values='Z').values
pip install numpy matplotlib
命令进行安装。contourf
函数来绘制等高线图。你可以根据需要调整颜色映射和其他参数。通过以上步骤,你可以使用numpy或pandas生成带有Z值的2D网格,并进行可视化展示。
领取专属 10元无门槛券
手把手带您无忧上云