Matplotlib legend位置设置|极客教程

Matplotlib legend位置设置

在使用matplotlib进行数据可视化时，经常需要在图表中添加图例来说明数据的含义。在matplotlib中，可以通过设置legend的位置来控制图例的显示位置。本文将介绍几种常用的图例位置设置方式，并提供相应的示例代码和效果展示。

默认位置

在matplotlib中，如果不指定legend的位置，系统会自动选择一个合适的位置进行显示。通常情况下，系统会尽可能地将legend放置在不影响数据展示的位置，比如右上角或左上角。

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 2, 3, 4, 5]

plt.plot(x, y1, label='y1')
plt.plot(x, y2, label='y2')
plt.legend()
plt.show()

Output:

Matplotlib legend位置设置

运行以上代码，将会显示一个具有默认图例位置的折线图。

显示在右上角

如果想将legend显示在右上角，可以通过设置loc='upper right'来实现。

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 2, 3, 4, 5]

plt.plot(x, y1, label='y1')
plt.plot(x, y2, label='y2')
plt.legend(loc='upper right')
plt.show()

Output:

Matplotlib legend位置设置

运行以上代码，将会显示一个将图例位置设置在右上角的折线图。

显示在左下角

同样，如果想将legend显示在左下角，可以通过设置loc='lower left'来实现。

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 2, 3, 4, 5]

plt.plot(x, y1, label='y1')
plt.plot(x, y2, label='y2')
plt.legend(loc='lower left')
plt.show()

Output:

Matplotlib legend位置设置

运行以上代码，将会显示一个将图例位置设置在左下角的折线图。

显示在左上角

除了右上角和左下角，还可以将legend显示在左上角，通过设置loc='upper left'来实现。

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 2, 3, 4, 5]

plt.plot(x, y1, label='y1')
plt.plot(x, y2, label='y2')
plt.legend(loc='upper left')
plt.show()

Output:

Matplotlib legend位置设置

运行以上代码，将会显示一个将图例位置设置在左上角的折线图。

显示在右下角

最后，也可以将legend显示在右下角，通过设置loc='lower right'来实现。

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 2, 3, 4, 5]

plt.plot(x, y1, label='y1')
plt.plot(x, y2, label='y2')
plt.legend(loc='lower right')
plt.show()