python绘制河流三维图

在Python中，你可以使用matplotlib库结合numpy和trimesh等第三方库来绘制三维河流图。以下是一个简单的示例步骤：

首先，安装必要的库：

pip install matplotlib numpy trimesh

然后，你可以使用以下代码生成并绘制一个基本的河流模型：

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
from trimesh.creation import extrude_polygon

# 创建河流数据
width = 5
depth = 0.5
poly = [(0, 0), (width, 0), (width, depth), (0, depth)]
vertices, faces = extrude_polygon(poly, height=1)  # 河床形状为矩形，高度为1单位

# 添加流线
dx, dy = 0.1, 0.1  # 流速方向和大小
streamlines = np.mgrid[0:width:dx, 0:depth:dy]
x, y = streamlines.reshape(2, -1).T
u, v = np.random.randn(len(x)), np.random.randn(len(y))  # 随机流速向量
vertices += [x + u * dx, y + v * dy]  # 更新点的位置

# 创建3D轴
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 绘制河床和流线
mesh = trimesh.Trimesh(vertices=vertices, faces=faces)
ax.plot_trisurf(mesh.vertices[:, 0], mesh.vertices[:, 1], mesh.vertices[:, 2],
                color='blue', alpha=0.5)
ax.streamplot(x, y, u, v, density=1, linewidth=0.5, color='red')

# 设置视口
ax.set_xlim([0, width])
ax.set_ylim([0, depth])
ax.set_zlim([0, 1])

plt.show()

这个例子生成了一个简单的河流模型，有固定的宽度、深度和随机分布的流线。你可以根据需要调整流速向量、流线密度和其他参数。