Future's blog
0%

Matplotlib Quick Reference

Posted on Edited on In

Cheatsheet

来自mpl官网的 https://matplotlib.org/cheatsheets/cheatsheets.pdf 的官方cheatsheet，但是pdf不方便即时查看，遂导出成图片，可以存到相册。图片可以单击后放大查看。

page1:

Cheatsheet

page2:

Cheatsheet

这里面覆盖了大部分使用mpl时会用到的函数，详细使用方法可见官网（https://matplotlib.org/）。

本文后续将会挑一些进行额外解释。

colormap class

colormap是把图变好看的灵魂。属于matplotlib.colors.Colormap()类. 这个类实现了__call__()方法，因此可以传入一个介于0-1之间的数，其返回一个四元组，表示一个RGBA颜色。这个元组可以被mpl的各类画图函数接收。

基础用法：

1
2
3
4
5
6
7
8
9
10
11
12
import matplotlib.pyplot as plt

cmap = plt.get_cmap('magma')

print(cmap(0))
print(cmap(0.5))
print(cmap(1))

# OUTPUT:
(0.001462, 0.000466, 0.013866, 1.0)
(0.716387, 0.214982, 0.47529, 1.0)
(0.002258, 0.001295, 0.018331, 1.0)

实践用法：画一条渐变的线。官网有例子 https://matplotlib.org/stable/gallery/lines_bars_and_markers/multicolored_line.html#sphx-glr-gallery-lines-bars-and-markers-multicolored-line-py, 我这是另一种实现方法。

1
2
3
4
5
6
7
8
9
10
11
12
13
import matplotlib.pyplot as plt
import numpy as np

cmap = plt.get_cmap('magma')

t = np.linspace(0, 10, 100)
xt = np.sin(t)
normalized_t = t / t.max() # make it into [0, 1]

for i in range(len(t)-1): # draw piece by piece
    plt.plot(t[i:i+2], xt[i:i+2], color=cmap(normalized_t[i]))

plt.show()

效果：

Cheatsheet