ax函数将两组数据画在一张图里，如何添加图例

可以使用 matplotlib 库中的 plt.legend() 函数来添加图例。只需要在函数中传入每组数据的 label 参数作为一个列表即可。例如：

import matplotlib.pyplot as plt

x1 = [1, 2, 3]
y1 = [4, 5, 6]
x2 = [1, 2, 3]
y2 = [7, 8, 9]

plt.plot(x1, y1, label='data1')
plt.plot(x2, y2, label='data2')

plt.legend()
plt.show()

这样就会在图中添加一个图例，分别标注为 data1 和 data2。