ax.legend()

这是一个用于在matplotlib中添加图例的函数。当你在画图时，可能会有多个线条或散点图，用图例可以让你更好地辨认它们。在调用这个函数之前，你需要先给每个线条或散点图添加一个label参数，这个参数是一个字符串，表示该线条或散点图的名字。然后，调用ax.legend()就可以在图表上添加一个图例，其中包含了每个线条或散点图的label。

相关问题

legend1 = ax.legend(*scatter.legend_elements()

这行代码是用来创建图例（legend）的。它使用了 Matplotlib 中的 `legend()` 函数，其中 `*scatter.legend_elements()` 表示将散点图中的每个不同的标签（label）作为一个元素传入，然后使用 `ax.legend()` 将它们合并为一个图例对象并添加到图像中。具体来说，`scatter.legend_elements()` 返回一个包含两个元素的元组，第一个元素是一个列表，包含每个标签对应的散点图中的所有元素，第二个元素是一个列表，包含每个标签的文本标签。因此，这个代码行将 `scatter.legend_elements()` 的返回值解包成两个变量，并将它们传递给 `ax.legend()` 函数来创建图例。

后续对每个子图设置legend 应该用plt.legend还是ax1.legend

如果使用plt.plot绘制图形，则使用plt.legend函数来设置图例；如果使用ax1.plot绘制图形，则使用ax1.legend函数来设置图例。

plt.legend函数是用来为全局figure设置图例的，它会在当前的figure上查找所有的已命名图形对象，并自动为它们创建图例。而ax1.legend函数则是用来为指定的子图对象设置图例的，它只会在指定的子图上查找已命名的图形对象，并创建图例。

如果你想为整个图形设置一个图例，可以使用plt.legend函数，如果你想为每个子图设置不同的图例，可以使用ax1.legend函数。但是需要注意的是，如果在使用ax1.plot绘制图形时已经为每个子图对象设置了名称，那么在使用plt.legend函数时也会自动为每个子图对象创建相应的图例。