Matplotlib入门

via: 保罗的酒吧 https://paul.pub/matplotlib-basics/

入门代码示例

import matplotlib.pyplot as plt
import numpy as np

data = np.arange(100,201)
plt.plot(data)
plt.show()
image.png
x = np.linspace(0, 2*np.pi,50)  # 50个元素均匀分布在[0,2pi]上
plt.plot(x, np.sin(x))
plt.show()
image.png

一次绘制多个图形

有些时候,我们可能希望一次绘制多个图形,例如:两组数据的对比,或者一组数据的不同展示方式等。

多个figure

可以简单的理解为一个figure就是一个图形窗口。matplotlib.pyplot会有一个默认的figure,我们也可以通过plt.figure()创建更多个。如下面的代码所示:

import matplotlib.pyplot as plt
import numpy as np

data = np.arange(100,201)
plt.plot(data)

data2 = np.arange(200,301)
plt.figure()
plt.plot(data2)

plt.show()
image.png

多个subplot

有些情况下,我们是希望在同一个窗口显示多个图形。此时就这可以用多个subplotsubplot函数的前两个参数指定了subplot数量,即:它们是以矩阵的形式来分割当前图形,两个整数分别指定了矩阵的行数和列数。而第三个参数是指矩阵中的索引。 下面是一段代码示例:

import matplotlib.pyplot as plt
import numpy as np

data = np.arange(100,201)
plt.subplot(2,1,1)
plt.plot(data)

data2 = np.arange(200,301)
plt.subplot(2,1,2)
plt.plot(data2)

plt.show()
image.png

subplot函数的参数不仅仅支持上面这种形式,还可以将三个整数(10之内的)合并一个整数。例如:2, 1, 1可以写成211,2, 1, 2可以写成212。

subplot函数的详细说明参见这里:matplotlib.pyplot.subplot

在一张图上绘制多个数据集

x = np.linspace(0, 2*np.pi,150)
plt.plot(x, np.sin(x), x, np.sin(2*x), x, np.sin(3*x))
plt.show()
image.png
x = np.linspace(0, 2*np.pi, 50)
plt.plot(x, np.sin(x), 'r-^', x, np.sin(2*x), 'g--')
plt.show()
image.png

自定义图形的外观

上述代码展示了两种不同的曲线样式:'r-o''g--'。字母 'r''g' 代表线条的颜色,后面的符号代表线和点标记的类型。例如 '-o' 代表包含实心点标记的实线,'--' 代表虚线。

颜色: 蓝色 - 'b' 绿色 - 'g' 红色 - 'r' 青色 - 'c' 品红 - 'm' 黄色 - 'y' 黑色 - 'k''b'代表蓝色,所以这里用黑色的最后一个字母) 白色 - 'w'

线: 直线 - '-' 虚线 - '--' 点线 - ':' 点划线 - '-.'

常用点标记 点 - '.' 像素 - ',' 圆 - 'o' 方形 - 's' 三角形 - '^'

https://matplotlib.org/api/markers_api.html

常用图形实例

Matplotlib可以生成非常多的图形式样,多到令人惊叹的地步。

https://matplotlib.org/gallery/index.html

线性图

前面的例子中,线性图的横轴的点都是自动生成的,而我们很可能希望主动设置它。另外,线条我们可能也希望对其进行定制。看一下下面这个例子:

import matplotlib.pyplot as plt
plt.plot([1,2,3],[3,6,9],'r')  # red
plt.plot([1,2,3],[2,4,9],':g')  # :(点) green

plt.show()
image.png

散点图

scatter函数用来绘制散点图。同样,这个函数也需要两组配对的数据指定x和y轴的坐标。下面是一段代码示例:

N = 20  # 点的个数

plt.scatter(np.random.rand(N)*100,
           np.random.rand(N)*100,
           c='r',s=100,alpha=0.5)  # c:点的颜色  s:点的大小  alpha:点的透明度

plt.scatter(np.random.rand(N)*100,
           np.random.rand(N)*100,
           c='g',s=200,alpha=0.5)

plt.scatter(np.random.rand(N)*100,
           np.random.rand(N)*100,
           c='b',s=300,alpha=0.5)

plt.show()
image.png

饼状图

pie函数用来绘制饼状图。饼状图通常用来表达集合中各个部分的百分比。

<bdi style="box-sizing: border-box;">In</bdi> [27]:

labels = ['Mon','Tue','Wed','Thu','Fri','Sat','Sun']

data = np.random.rand(7)*100

plt.pie(data, labels=labels, autopct='%1.1f%%')  # autopct 指定数值的精度
plt.axis('equal')  # 设置坐标轴大小一致
plt.legend()  # 指明要绘制图例

plt.show()
image.png

条形图

bar函数用来绘制条形图。条形图常常用来描述一组数据的对比情况,例如:一周七天,每天的城市车流量。

N = 7

x = np.arange(7)
data = np.random.randint(low=0, high=100, size=N)  # 生成7个0-100的随机数
colors = np.random.rand(N*3).reshape(N,-1)  # 先生成21(N x 3)个随机数,然后将它们组装成7行,那么每行就是三个数,这对应了颜色的三个组成部分。
labels = ['Mon','Tue','Wed','Thu','Fri','Sat','Sun']

plt.title('Weekday Data')
plt.bar(x, data, alpha=0.8, color=colors, tick_label=labels)
plt.show()
image.png

直方图

hist函数用来绘制直方图。直方图看起来是条形图有些类似。但它们的含义是不一样的,直方图描述了数据中某个范围内数据出现的频度。

data = [np.random.randint(0,n,n) for n in [3000,4000,5000]]
labels = ['3K','4K','5K']
bins = [0,100,500,1000,2000,3000,4000,5000]

plt.hist(data, bins=bins, label=labels)
plt.legend()
plt.show()

image.png

上面的这些代码中,[np.random.randint(0, n, n) for n in [3000, 4000, 5000]]生成了包含了三个数组的数组,这其中:

  • 第一个数组包含了3000个随机数,这些随机数的范围是 [0, 3000)
  • 第二个数组包含了4000个随机数,这些随机数的范围是 [0, 4000)
  • 第三个数组包含了5000个随机数,这些随机数的范围是 [0, 5000)
    bins数组用来指定我们显示的直方图的边界,即:[0, 100) 会有一个数据点,[100, 500)会有一个数据点,以此类推。所以最终结果一共会显示7个数据点。同样的,我们指定了标签和图例。

在这幅图中,我们看到,三组数据在3000以下都有数据,并且频度是差不多的。但蓝色条只有3000以下的数据,橙色条只有4000以下的数据。这与我们的随机数组数据刚好吻合。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,591评论 6 501
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,448评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,823评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,204评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,228评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,190评论 1 299
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,078评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,923评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,334评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,550评论 2 333
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,727评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,428评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,022评论 3 326
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,672评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,826评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,734评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,619评论 2 354

推荐阅读更多精彩内容