如何用Matplotlib可视化含Z坐标的赛车赛道数据（展示高度差异）

针对你遇到的可视化问题，以下几个Matplotlib方案能更直观展示赛道的高度差异：

1. 带高度着色的2D赛道投影图 #

在XY平面的赛道路径上，用颜色映射Z轴高度，既保留赛道的平面走向，又能通过颜色直观区分高低，适合快速定位赛道陡坡/平缓段：

import matplotlib.pyplot as plt
import numpy as np

# 替换为你的真实X/Y/Z数据集
X = np.random.uniform(0, 100, 500)
Y = np.random.uniform(0, 100, 500)
Z = np.sin(X/20) * np.cos(Y/20) * 20 + np.random.normal(0, 2, 500)

plt.figure(figsize=(10,8))
# 浅色折线保留赛道连贯性，散点颜色绑定Z值
plt.plot(X, Y, c='black', alpha=0.3, zorder=1)
scatter = plt.scatter(X, Y, c=Z, cmap='viridis', s=10, zorder=2)
# 添加颜色条标注高度数值
plt.colorbar(scatter, label='赛道高度(Z)')
plt.xlabel('X坐标')
plt.ylabel('Y坐标')
plt.title('赛车赛道平面投影+高度着色')
plt.show()

2. 侧视角3D赛道折线图（带高度着色） #

调整3D视角到赛道侧上方，同时用颜色映射Z值，解决普通3D折线重叠、高度不明显的问题：

fig = plt.figure(figsize=(12,8))
ax = fig.add_subplot(111, projection='3d')
# 3D折线+散点，颜色均绑定Z值
ax.plot(X, Y, Z, c='black', alpha=0.2)
scatter = ax.scatter(X, Y, Z, c=Z, cmap='plasma', s=15)
# 调整视角为侧上方，更易观察起伏
ax.view_init(elev=30, azim=45)
ax.set_xlabel('X坐标')
ax.set_ylabel('Y坐标')
ax.set_zlabel('高度(Z)')
plt.colorbar(scatter, label='赛道高度(Z)')
plt.title('侧视角3D赛道+高度着色')
plt.show()

3. 赛道高度热力图 #

把Z值转换成XY平面的热力图，再叠加赛道路径，适合全局观察赛道高度分布：

from matplotlib.colors import Normalize
from scipy.interpolate import griddata

# 替换为你的真实X/Y/Z数据集
X = np.random.uniform(0, 100, 500)
Y = np.random.uniform(0, 100, 500)
Z = np.sin(X/20) * np.cos(Y/20) * 20 + np.random.normal(0, 2, 500)

# 创建XY网格并插值Z值，生成平滑热力图
xi, yi = np.meshgrid(np.linspace(X.min(), X.max(), 100), np.linspace(Y.min(), Y.max(), 100))
zi = griddata((X, Y), Z, (xi, yi), method='cubic')

plt.figure(figsize=(10,8))
# 绘制高度热力图
heatmap = plt.imshow(zi, extent=[X.min(), X.max(), Y.min(), Y.max()], origin='lower', cmap='coolwarm', alpha=0.7)
# 叠加白色赛道路径
plt.plot(X, Y, c='white', linewidth=2)
plt.colorbar(heatmap, label='赛道高度(Z)')
plt.xlabel('X坐标')
plt.ylabel('Y坐标')
plt.title('赛道高度热力图+路径叠加')
plt.show()

内容的提问来源于stack exchange，提问作者Liya