[PYTHON]希望在散点图(棒球)上绘制发射速度与发射角度的关系，但我不知道如何使用Matplotlib根据间距类型对它们进行着色

import matplotlib.pyplot as plt
import numpy as np

# 发射速度和发射角度数据
velocity = [10, 15, 20, 25, 30]
angle = [30, 45, 60, 75, 90]

# 创建散点图
plt.scatter(velocity, angle)

# 根据间距类型进行着色
colors = np.sqrt(np.power(velocity, 2) + np.power(angle, 2))
plt.scatter(velocity, angle, c=colors, cmap='viridis')

# 添加颜色条
cbar = plt.colorbar()
cbar.set_label('Distance')

# 添加标题和坐标轴标签
plt.title('Relationship between Launch Velocity and Launch Angle')
plt.xlabel('Launch Velocity')
plt.ylabel('Launch Angle')

# 显示图形
plt.show()