matplotlib画图总结--常用功能

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)）