首页 > 编程知识 正文

Python中plot加图例

时间:2023-11-19 15:12:25 阅读:295050 作者:YPYM

图例(legend)是用于解释图表中各个元素的标识,使得读者能够更好地理解图表的含义。在Python中,我们可以使用matplotlib库的plot函数来绘制图表,并通过添加图例来说明每个数据系列的含义。

一、使用plot函数绘制图表

我们首先需要了解如何使用plot函数来绘制图表。plot函数接受两个参数,分别是x轴和y轴的数据序列。下面是一个简单的例子:

import matplotlib.pyplot as plt

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

plt.plot(x, y)
plt.show()

运行以上代码,会得到一个简单的折线图。x轴为1, 2, 3, 4, 5,y轴为1, 4, 9, 16, 25。

二、添加图例

接下来,我们需要为图表中的每个数据系列添加图例。我们可以通过给plot函数传入label参数来指定每个数据系列的标签,然后通过调用legend函数来添加图例。

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 8, 27, 64, 125]

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

plt.legend()
plt.show()

运行以上代码,会得到一个带有图例的折线图。其中,y1的图例标签为'y1',y2的图例标签为'y2'。

三、自定义图例位置

有时候,我们可能需要将图例放置在图表的不同位置。我们可以通过传入loc参数来指定图例的位置,loc参数可以接受以下字符串参数:

  • 'best':自动选择最佳位置
  • 'upper right':右上角
  • 'upper left':左上角
  • 'lower right':右下角
  • 'lower left':左下角
  • 'center right':右边中间
  • 'center left':左边中间
  • 'center':中间
import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 8, 27, 64, 125]

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

plt.legend(loc='upper right')
plt.show()

运行以上代码,会得到一个将图例放置在右上角的折线图。

四、自定义图例样式

我们还可以通过传入prop参数来自定义图例的样式。prop参数可以接受字体设置的相关参数,比如fontsize、weight、style等。

import matplotlib.pyplot as plt
from matplotlib.font_manager import FontProperties

x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 8, 27, 64, 125]

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

font = FontProperties()
font.set_size('large') # 设置字体大小

plt.legend(prop=font)
plt.show()

运行以上代码,会得到一个字体大小为'large'的折线图图例。

五、总结

本文详细介绍了如何在Python中使用matplotlib库的plot函数绘制图表,并通过添加图例来说明每个数据系列的含义。我们学习了如何使用label参数设置每个数据系列的标签,以及如何使用legend函数添加图例。此外,我们还介绍了如何自定义图例的位置和样式。

通过合理利用图例,我们可以使得图表更加清晰易读,提高数据可视化的效果。

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。