Python 3D绘图中Quiver矢量箭头分辨率提升方法咨询

问题核心原因 #

你当前代码中箭头分辨率极低的关键问题是箭头数量过少：通过两层for循环仅生成了(4+1)*(4+1)=25个箭头，完全无法体现场线的分布细节。此外，部分参数设置和曲面计算逻辑也存在可优化空间。

具体改进措施 #

• 大幅提升箭头密度：放弃固定循环次数的方式，改用np.meshgrid生成高密度网格点，根据需求调整点数（如20×20或50×50），批量绘制箭头。
• 优化Quiver显示参数：调整length（箭头长度）、linewidth（箭杆粗细）、headwidth（箭头头部宽度）等参数，让箭头轮廓更清晰；若无需归一化场矢量，可去掉normalize=True避免不必要的缩放。
• 修复曲面计算的潜在错误：原代码中zz = (xx+yy)/(xx+yy) + z_z -1在xx+yy=0时会触发除以0警告，改用np.where处理边界情况，同时简化冗余的z轴逻辑。
• 统一绘图DPI设置：创建画布时就指定DPI，确保显示窗口和保存图像的清晰度一致。

修改后的完整代码 #

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits import mplot3d

angle = 0 

b_y = np.cos(angle*np.pi/180) 
b_z = np.sin(angle*np.pi/180) 

l = np.array([1,0,0])   
b = np.array([0,b_y,b_z])   
A = np.cross(l,b)       

l_ = np.linalg.norm(l)  
b_ = np.linalg.norm(b)  
A_ = np.linalg.norm(A)  

# 创建画布时指定DPI，提升显示清晰度
plt.figure(1, dpi=100)
ax = plt.axes(projection="3d")
ax.set_xlabel("x")
ax.set_ylabel("y")
ax.set_zlabel("z")

ax.set_xlim(0, 1)
ax.set_ylim(0, 1)
ax.set_zlim(0, 1)

# 绘制曲面 - 修复除以0问题并简化逻辑
n = 100 
x_l = np.linspace(0, b[1], n)
y_b = np.linspace(0, 1, n)
xx, yy = np.meshgrid(x_l, y_b)
# 用np.where处理xx+yy=0的边界情况，简化z轴计算
zz = np.where(xx + yy != 0, (xx+yy)/(xx+yy), 1) + b[2] - 1  
ax.plot_surface(xx, yy, zz, alpha=0.5)

B = np.array([0,0,1]) # 场矢量
flux = np.dot(A,B) # 通量

# 设置箭头密度，生成高密度网格点
quiv_density = 20
x_points = np.linspace(0, 1, quiv_density)
y_points = np.linspace(0, 1, quiv_density)
xx_quiv, yy_quiv = np.meshgrid(x_points, y_points)
zz_quiv = np.zeros_like(xx_quiv)  # 箭头起始z坐标设为0，可按需贴合曲面调整

# 批量绘制箭头，优化显示参数
ax.quiver(yy_quiv, xx_quiv, zz_quiv, 
          B[0], B[1], B[2], 
          length=0.05, linewidth=0.4, headwidth=3, 
          color='purple')

plt.savefig('3Dplot1.png', dpi=300)
plt.show()

额外提示 #

• 可根据设备性能和需求调整quiv_density数值：数值越大箭头越密集，但会增加绘图耗时。
• 如果需要箭头贴合曲面，可将zz_quiv替换为曲面对应位置的z值（直接使用曲面的zz网格数据即可）。

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