箭袋3d (python matplotlib)：如何让箭杆和箭头获得相同的颜色？(震颤虫？)

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

x_start, y_start, z_start = 0, 0, 0
x_end, y_end, z_end = 1, 1, 1

arrow = Arrow3D([x_start, x_end], [y_start, y_end], [z_start, z_end], color='red')

ax.add_artist(arrow)

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

plt.show()