用Python画如此漂亮的专业插图 ?简直So easy!

本文介绍如何利用Python的matplotlib库创建专业级别的图表,包括2D/3D图形、Scatter图、3D数据可视化、矢量场和网络图,以及LaTeX公式插入,同时提及了与Matlab的对比和个性化配色技巧。

本文整理自知乎问答,仅用于学术分享,著作权归作者所有。如有侵权,请联系后台作删文处理。

编译:极市平台

方法一

作者|冯昱尧

https://www.zhihu.com/question/21664179/answer/18928725

强烈推荐 Python 的绘图模块 matplotlib: python plotting 。画出来的图真的是高端大气上档次,低调奢华有内涵~ 适用于从 2D 到 3D,从标量到矢量的各种绘图。能够保存成从 eps, pdf 到 svg, png, jpg 的多种格式。并且 Matplotlib 的绘图函数基本上都与 Matlab 的绘图函数名字都差不多,迁移的学习成本比较低。开源免费。如图所示(题目描述中的图在最后):(以下图片均引用自 Thumbnail gallery )

像这种普通的函数图象:

00f937c00a55f130bc6a89d535a0bf27.png

plt.fill(x, y1, 'b', x, y2, 'r', alpha=0.3)


以及这种 Scatter 图(中文不知道该怎么说…):

eae3dc29eea5d67fd90702d2b18b5922.png

plt.scatter(x, y, s=area, alpha=0.5)

精致的曲线,半透明的配色。显出你那高贵冷艳的X格,最重要的是只需一行代码就能搞定。从此再也不用忍受 Matlab以及GNUPlot 中那蛋疼的配色了。

想画 3D 数据?没有问题 (用mayavi可能更方便):

e877787d0566d7cd4d872d010d124a40.png

ax.plot_surface(X, Y, Z, rstride=8, cstride=8, alpha=0.3)
cset = ax.contourf(X, Y, Z, zdir='z', offset=-100, cmap=cm.coolwarm)
cset = ax.contourf(X, Y, Z, zdir='x', offset=-40, cmap=cm.coolwarm)
cset = ax.contourf(X, Y, Z, zdir='y', offset=40, cmap=cm.coolwarm)

四行代码你就能拥有(后三行是画坐标平面上的等高线,严格的额说还是一行)。

除此以外,不过你是矢量场,网络还是什么奇葩的需求都能够搞定:

ce0f006528468349e7b186efcd557455.png

plt.streamplot(X, Y, U, V, color=U, linewidth=2, cmap=plt.cm.autumn)
plt.colorbar()

f7a673fea9fafd12cd01097cb50e3664.png

plt.triplot(x, y, triangles, 'go-')
plt.title('triplot of user-specified triangulation')
plt.xlabel('Longitude (degrees)')
plt.ylabel('Latitude (degrees)')

832c1db78ea58d227012ea1dd2acddd0.png

ax = plt.subplot(111, polar=True)
bars = ax.bar(theta, radii, width=width, bottom=0.0)

这还没完,Matplotlib 还支持Latex公式的插入,当别人画的图还是这个样子的时候(以下图片引用自Matplotlib Tutorial(译))

a323d14fd339cb677f7c3d91f4ded74f.png

你能够把它变成这个样子:

0ec7ab44c5e3f51efb83fa69b7728ec0.png

如果再搭配上 IPython 作为运行终端(这张图是自己绘制的~):

8d5f9fc0c4f08e352a1a6a8dff4a7363.png


简直就是神器啊,有木有!

心动不如行动,还等什么?

经@许铖同学提醒,再补充一句,matplotlib 还可以话 xkcd 风格的图呦~

5a2e3a45ab6559b0e872189c22085ba8.png

(图片引用自网络)

此外结合 IPython Notebook 后更多精彩内容,请看http://nbviewer.ipython.org/

如果嫌安装麻烦并且恰好在 Windows 系统下的话可以尝试Python的一个发行版winpython - Portable Scientific Python 2/3 32/64bit Distribution for Windows。

鉴于@van li同学质疑 matplotlib 是否能画出题目中所示的图像,我在这里将题目中的图像用 matplotlib 画出来如下:

f52a1be994b70286450330a390cf68b7.png

代码在此处:

https://gist.github.com/coldfog/c479124328fc6bb8b789

6cb2248484c452e0b438137da7e8f46b.png

代码在此处:

https://gist.github.com/coldfog/5da63a6958fc0a949b52

看到楼下有人说配色和好看,唉....那我也贴几个吧...只不过当初限于篇幅没有写而已。

首先,python有一个专门的配色包jiffyclub/brewer2mpl,提供了从美术角度来讲的精美配色(戳这里感受ColorBrewer: Color Advice for Maps)。


此外还有一些致力于美化绘图的库,用起来也都非常方便,比如olgabot/prettyplotlib 。

废话不多说,上图就是王道。(下面图片来源网络)

6392d0c275be4d0ab610d35f25184d45.png

432582d237ce2d96dfe629d80df0a1e2.png

1aade663ff6a41720e607ed6f51563aa.png


有人可能会说需要复杂的设置,其实也不用。比如上边这幅图,只需要多加一个参数就好:

cmap=brewer2mpl.get_map('RdBu', 'diverging', 8, reverse=True).mpl_colormap,


楼下说到统计绘图。嘛seaborn 是一个调用 matplotlib 的统计绘图库,上图:

(https://github.com/mwaskom/seaborn) 

756fb85b236637c61aac41c6d567036a.png

代码一行,后边的几乎都是一行,没做其他设置,默认就这样。我就不贴其他的代码了:

g = sns.jointplot(x1, x2, kind="kde", size=7, space=0)

8d02d02fbe0d8dea6715aba006587480.png

e1d9faa35330c4f590e4f67c5f18a4c8.png

28f617bddceffc880f633ee2c4dd1355.png

e2367c21a93596143f4d676c4b6109d7.png

d20b154db7da579d8fc4f4091bdb6cf7.png

还有个更炫酷的可交互式绘图,大家自己戳开看吧:

http://nbviewer.ipython.org/github/plotly/python-user-guidechaocc/blob/master/s0_getting-started/s0_getting-started.ipynb

哼哼,完爆了吧~~~~\(≧▽≦)/~

---

遇到安装问题的请尝试Anaconda这个Python发行版。下载安装后直接使用即可,它几乎预装了所有要用到的科学计算及可视化的库。

有盆友在评论里说希望能有完整的教程,确实就这个答案来说,离实际使用还有很大的距离,网上相关的中文资料也不多。不过真要写起来这个答案也装不下,况且写在这个问题下也不是很恰当。等到那天我有专栏了再说吧,到时候也许会写一个关于可视化的系列教程。

方法二

作者|阿昆

https://www.zhihu.com/question/21664179/answer/1182984311

翻遍这个问题下的所有回答,发现凡是提到Matlab的,其评价中常有‘锯齿’,‘菜鸟’,‘难看’,‘不忍直视’等标签。

然而,2020年了,技术提升了,观念进步了,当一些基本问题解决后,Matlab还那么‘不堪’吗?

2a96c3c6f6a2159f7be99baea62f3c85.png

观察Mathematica、Origin、Python/matplotlib、R/ggplot2等软件绘制的数据、结果图,其与Matlab图的差异主要体现在点、线、面等对象属性(位置、尺寸、颜色等)的不同上

既然只是属性的不同,那是不是只要修改一下这些信息,就可以实现各种软件绘图风格之间的转换了呢?

答案是肯定的。

比如,这是高赞回答 @冯昱尧用Python/matplotlib绘制的一幅图:

7aaa7e5ffdc14fe596fd842dee61d5bd.png

我们用Matlab默认属性来绘制,效果是这样的(没加误差棒):

c32f7058b33591ef9850825882dbf227.png

然后,只需再修改一下位置、尺寸、颜色等信息,就可以得到风格差不多的图(没加误差棒):

da826f1657b239351663e1d6f337a034.png

当我们用这一思想来思考该如何绘制插图时,就很容易实现自己的小想法,仿造甚至创造出理想的插图。

比如,某一天,发现傍晚的天空颜色很美,心想:为什么不能把它画到论文插图里呢?(见:Matlab论文插图配色2——自然渐变)

于是,

5f56d4f167ac3f77a500394919195e17.png

8e90bac2c5c1e7c843747dc716f7542b.png

再比如,某一天,看到女朋友的照片,觉得很美,心想:为什么不能把她画到论文插图里呢?(见:Matlab论文插图配色1——是女朋友的颜色)

于是,

d47fbe5b6646d30977fc60cf63807998.png

这时,有朋友就要说了:“哎呀答主,你整这些个花里花哨的东西,还不是得一行代码一行代码的敲出来啊,太麻烦了吧。”

此言差矣。

就像R有ggplot2,Python有matplotlib,Matlab其实也有很多现成的绘图工具包,并不需要你自己开发。

比如,

Pierre Morel [1] 结合ggplot2,开发了gramm工具,用于绘制复杂图形。

Inspired by ggplot2 (Wickham 2009), the R implementation of “grammar of graphics” principles (Wilkinson 1999), gramm improves Matlab’s plotting functionality, allowing to generate complex figures using high-level object-oriented code.

示例效果如下:

e3ebaea59536f4c4a0f4c55e8c6634ae.png

b8eaef0dbcbc0da8d2b7ced885cd92bb.png

8a923b083334e9028154ef6ba08e3bdf.png

afc25329404af56b87da45ed42421626.png

423b56759e990d730ea0a66b9e9648f0.png

cc8ded5c921a2b4e9d76741b8562f7f9.png

8dc3e07a8cc21614ff5fcf322d61c39c.png

4d7958debf5523349a8a0878e1e739e7.png

45aab246349ff6ea2af9960c636c6c69.png

dac04cb5bae42f80f07217761bddd7e2.png

fad46f33ab187dab66daed75b5f3e4de.png

类似的,Stephen Cobeldick [2] 将matplotlib配色方案移植到了Matlab。

也就是说,在Matlab中就可以直接用matplotlib的配色方案了,就不必总是‘jet’了。

The MatPlotLib 2.0 default colormaps ported to MATLAB. This submission also includes the Line ColorOrder colormaps!

示例效果如下:

33fd38338ea88ad2d4409031d7b8c4e8.png

31474f46abf56f750e2de41242b1f793.png

还有很多专门针对论文插图的工具包,这里就不一一介绍了。

总的来说,工具只是工具,它们并没有高低贵贱之分。

若想画出好看的插图,关键还是在于使用工具的人

集中一点,登峰造极。

参考:
Morel P . Gramm: grammar of graphics plotting in Matlab.
Cobeldick S . MatPlotLib Perceptually Uniform Colormaps.

---------End---------

顺便给大家推荐下我的微信视频号「Python数据之道」,欢迎扫码关注。

6060838a126201a2b829c89dda4156cd.gif

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值