Python显示图例代码
本文将从如何绘制图例、如何修改图例颜色、如何添加标题和标签等多个方面详细阐述Python显示图例代码,帮助大家更加灵活自如地运用图例功能。
Python中,绘制图例时需要用到legend()函数。首先,我们需要在绘制图形时给每个数据集添加一个label标签,然后在legend()函数中传入这个label标签,即可自动绘制图例。
import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)
plt.plot(x, y1, label='sin(x)')
plt.plot(x, y2, label='cos(x)')
plt.legend()
plt.show()
以上代码中,我们使用NumPy生成了两个数据集y1和y2,并用plt.plot()函数绘制了它们的线性图。接着,我们在每个plt.plot()函数中传入一个label参数,用于绘制图例。最后,在plt.legend()函数中调用自动绘制图例功能即可。
默认情况下,Python会自动根据线条颜色来绘制图例。如果我们想要改变图例的颜色,可以在legend()函数中传入一个color参数指定颜色。
import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)
plt.plot(x, y1, label='sin(x)', color='blue')
plt.plot(x, y2, label='cos(x)', color='red')
plt.legend()
plt.show()
以上代码中,我们在plt.plot()函数中传入一个color参数指定每个数据集的颜色。最后,在plt.legend()函数中调用自动绘制图例功能即可。
除了绘制图例,在绘制图形时还需要添加标题和标签。我们可以使用plt.title()函数添加主标题,plt.xlabel()函数添加x轴标签,plt.ylabel()函数添加y轴标签。
import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)
plt.plot(x, y1, label='sin(x)', color='blue')
plt.plot(x, y2, label='cos(x)', color='red')
plt.title('Sin and Cos Waves')
plt.xlabel('x-axis')
plt.ylabel('y-axis')
plt.legend()
plt.show()
以上代码中,我们使用plt.title()函数、plt.xlabel()函数和plt.ylabel()函数添加主标题和坐标轴标签。最后,在plt.legend()函数中调用自动绘制图例功能即可。
有时候,我们需要在图例中设置更改字体大小和位置。我们可以使用plt.legend()函数的fontsize参数设置字体大小,使用loc参数设置位置。
import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)
plt.plot(x, y1, label='sin(x)', color='blue')
plt.plot(x, y2, label='cos(x)', color='red')
plt.title('Sin and Cos Waves')
plt.xlabel('x-axis')
plt.ylabel('y-axis')
plt.legend(fontsize='large', loc='upper right')
plt.show()
以上代码中,我们在plt.legend()函数中传入fontsize参数设置字体大小,传入loc参数设置位置。其中,位置可以设置为’best’、’upper right’、’lower left’等不同的字符串。
最后,如果默认的图例样式不能满足我们的需求,我们还可以自定义图例样式。我们可以通过创建一个新的legend对象,然后设置其属性来实现自定义图例样式。
import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)
plt.plot(x, y1, label='sin(x)', color='blue')
plt.plot(x, y2, label='cos(x)', color='red')
plt.title('Sin and Cos Waves')
plt.xlabel('x-axis')
plt.ylabel('y-axis')
legend = plt.legend(frameon=True)
frame = legend.get_frame()
frame.set_facecolor('#F0F0F0')
plt.show()
以上代码中,我们首先创建了一个legend对象,然后通过设置对象的属性实现自定义图例样式。具体来说,我们通过legend.get_frame()获取legend对象的方框对象frame,再通过frame.set_facecolor()设置方框颜色为灰色。
本文详细介绍了Python显示图例代码的多个方面,包括绘制图例、修改图例颜色、添加标题和标签等。希望本文能够帮助大家更好地了解和运用Python绘图功能。