绘制3D向量加法

def solve():
    import matplotlib.pyplot as plt
    import numpy as np
    from mpl_toolkits.mplot3d import Axes3D
    
    v1 = (4, 0, 3)
    v2 = (-1, 0, 1)
    v_sum = (v1[0]+v2[0], v1[1]+v2[1], v1[2]+v2[2])
    
    fig = plt.figure(figsize=(10, 8))
    ax = fig.add_subplot(111, projection='3d')
    
    ax.quiver(0, 0, 0, v1[0], v1[1], v1[2], color='red', arrow_length_ratio=0.1, label='v1=(4,0,3)')
    ax.quiver(v1[0], v1[1], v1[2], v2[0], v2[1], v2[2], color='blue', arrow_length_ratio=0.1, label='v2=(-1,0,1)')
    ax.quiver(0, 0, 0, v_sum[0], v_sum[1], v_sum[2], color='green', arrow_length_ratio=0.1, label=f'v1+v2={v_sum}')
    
    ax.set_xlim([-2, 5])
    ax.set_ylim([-2, 2])
    ax.set_zlim([-2, 5])
    ax.set_xlabel('X')
    ax.set_ylabel('Y')
    ax.set_zlabel('Z')
    ax.legend()
    plt.title('3D向量加法')
    plt.tight_layout()
    plt.show()
    
    print(f"向量和 = {v_sum}")