Python使用Matplotlib绘制三维曲面图

使用Python绘制三维曲面图

本教程将演示如何使用Python的Matplotlib库绘制三维曲面图。

1. 导入必要的库

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

• numpy 用于数值计算，例如创建和操作数组。
• pandas 用于数据处理，例如读取Excel文件。
• matplotlib.pyplot 用于创建图表。
• mpl_toolkits.mplot3d 用于创建三维图形。

2. 读取数据

data = pd.read_excel('C题数据.xlsx', sheet_name=2)
data = data.values

• 使用pd.read_excel函数读取名为'C题数据.xlsx'的Excel文件，并选择名为'2'的工作表。
• 使用.values属性将数据转换为NumPy数组。

3. 准备数据

x = data[:, 0]
y = data[:, 1]
z = data[:, 2]

x = x.flatten()
y = y.flatten()
z = np.reshape(z, [75, 1])
z = z.flatten()

• 将数据数组分成三列，分别赋值给x、y和z。
• 使用.flatten()方法将x、y和z转换为一维数组。
• 使用np.reshape()函数将z数组的形状更改为(75, 1)。
• 再次使用.flatten()方法将z转换为一维数组。

4. 创建三维图形

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

• 使用plt.figure()创建一个新的图形对象。
• 使用fig.add_subplot(111, projection='3d')创建一个三维坐标系。

5. 绘制三维曲面图

pic = ax.plot_trisurf(x, y, z)

• 使用ax.plot_trisurf()函数绘制三维曲面图，并将x、y和z作为输入数据。

6. 显示图形

plt.show()

• 使用plt.show()函数显示图形。

总结

本教程介绍了如何使用Python的Matplotlib库绘制三维曲面图。我们首先导入了必要的库，然后读取并准备了数据。接下来，我们创建了一个三维图形，并使用plot_trisurf()函数绘制了曲面图。最后，我们使用plt.show()函数显示了图形。

希望本教程对您有所帮助！