Python绘制水汽通量

import numpy as np  
import matplotlib.pyplot as plt  
  
# 假设数据  
# U, V 是风速分量（m/s），q 是比湿（g/kg）  
U = np.random.rand(10, 10) * 10  # 随机生成的水平风速分量  
V = np.random.rand(10, 10) * 5   # 随机生成的垂直风速分量  
q = np.random.rand(10, 10) * 20  # 随机生成的比湿  
  
# 计算水汽通量  
QU = U * q  # 水平水汽通量分量  
QV = V * q  # 垂直水汽通量分量  
  
# 绘制水汽通量  
plt.figure(figsize=(12, 6))  
  
# 绘制水平水汽通量分量  
plt.subplot(1, 2, 1)  
plt.contourf(QU, levels=50, cmap='Blues')  
plt.colorbar(label='Horizontal Water Vapor Flux (g/(m*s))')  
plt.title('Horizontal Water Vapor Flux')  
  
# 绘制垂直水汽通量分量  
plt.subplot(1, 2, 2)  
plt.contourf(QV, levels=50, cmap='Greens')  
plt.colorbar(label='Vertical Water Vapor Flux (g/(m*s))')  
plt.title('Vertical Water Vapor Flux')  
  
plt.tight_layout()  
plt.show()