0、内容范围
多曲线图、图例、坐标轴、注释文字等。
1、曲线图
多曲线图、图例、网格、坐标轴名称、图标名、坐标轴范围等。
from matplotlib import pyplot as plt import numpy as np x = np.linspace(-np.pi, np.pi, 200, endpoint=True) c, s = np.cos(x), np.sin(x) plt.xlim(-np.pi, np.pi) # p1 = plt.plot(x,c,'r', label = 'cos') # p2 = plt.plot(x,s,'b', label = 'sin') p1 = plt.plot(x,c,'r') p2 = plt.plot(x,s,'b') plt.xlabel('x') plt.ylabel('y') plt.title('cos and sin') plt.legend( ['cos', 'sin']) plt.grid(True) plt.show()
知识点:
1)一个图里绘制条曲线
方法1:
plt.plot(x,c,'r', x,s,'b')
方法2:
p1 = plt.plot(x,c,'r')
p2 = plt.plot(x,s,'b')
2)给图添加图例
方法1:
p1 = plt.plot(x,c,'r', label = 'cos')
p2 = plt.plot(x,s,'b', label = 'sin')
plt.legend()
方法2:
p1 = plt.plot(x,c,'r')
p2 = plt.plot(x,s,'b')
plt.legend( ['cos', 'sin'])
推荐使用方法1,因为方法2必须注意添加曲线的顺序。
2、柱状图
学习柱状图、为图形添加字符坐标轴.
import numpy as np import matplotlib.pyplot as plt N = 5 menMeans = (20, 35, 30, 35, 27) womenMeans = (25, 32, 34, 20, 25) menStd = (2, 3, 4, 1, 2) womenStd = (3, 5, 2, 3, 3) ind = np.arange(N) # the x locations for the groups width = 0.35 # the width of the bars: can also be len(x) sequence p1 = plt.bar(ind, menMeans, width, yerr=menStd, label='men') p2 = plt.bar(ind, womenMeans, width, bottom=menMeans, yerr=womenStd, label='women') plt.ylabel('Scores') plt.title('Scores by group and gender') plt.xticks(ind, ('G1', 'G2', 'G3', 'G4', 'G5')) plt.yticks(np.arange(0, 81, 10)) plt.legend(loc='upper right') plt.show()
为什么会两个图层叠?
p1 = plt.bar(ind, menMeans, width, yerr=menStd, label='men') p2 = plt.bar(ind, womenMeans, width, bottom=menMeans, yerr=womenStd, label='women')
bar函数的x坐标一样,并且y坐标值是bottom=menMeans,所以第二个图会堆叠在第一个图上面。
下面实现2条柱状图。
import matplotlib import matplotlib.pyplot as plt import numpy as np labels = ['G1', 'G2', 'G3', 'G4', 'G5'] men_means = [20, 34, 30, 35, 27] women_means = [25, 32, 34, 20, 25] x = np.arange(len(labels)) # the label locations width = 0.35 # the width of the bars fig, ax = plt.subplots() rects1 = ax.bar(x - width/2, men_means, width, label='Men') rects2 = ax.bar(x + width/2, women_means, width, label='Women') # Add some text for labels, title and custom x-axis tick labels, etc. ax.set_ylabel('Scores') ax.set_title('Scores by group and gender') ax.set_xticks(x) ax.set_xticklabels(labels) ax.legend() plt.show()
x轴的坐标不是数子,实现方法如下,先设定坐标轴的tick,然后把ticklabel改为字符值。
ax.set_xticks(x)
ax.set_xticklabels(labels)
如果需要实现水平柱状图,则使用ax.barh(y_pos, performance, xerr=error, align='center')函数。
3、添加文字、注释、箭头指示、花式文字
其中,text 函数可以做文本注释,且支持 LaTeX 格式,可以在图例中写公式。
text(x,y,string,fontsize=15,verticalalignment="top",horizontalalignment="right")
x = np.arange(0, 10, 0.1) plt.plot(x, x**2) plt.grid(True) # 设置网格线 plt.text(5,50, "TEXT1") plt.show()
花式文本:
You can put a rectangular box around the text instance (e.g., to set a background color) by using the keyword bbox. bbox is a dictionary of Rectangle properties. For example:
>>> text(x, y, s, bbox=dict(facecolor='red', alpha=0.5))
import matplotlib.pyplot as plt import numpy as np plt.text(0.6, 0.5, "Text", size=50, rotation=45, ha="center", va="center", color='b') plt.text(0.3, 0.5, "Text", size=25, rotation=10, ha="center", va="center", bbox=dict(boxstyle="round",ec=(1, 0.5, 0.5),fc=(1., 0.8, 0.8),)) #颜色ec、fc plt.plot() plt.show()
添加箭头和说明信息
plt.annotate('value of 5**2', xy=(5, 25), xytext=(6, 26), arrowprops=dict(facecolor='black', shrink=0.05))