在Python关于绘图，Mlab提供开源的matplotlib模块，不仅可以绘制折线图、柱状图、散点图等常规图外，还支持绘制量场图、频谱图、提琴图、箱型图等特殊图，例举往期文章可前往查看详情。

我们日常生活中经常会关注天气预报，在换季的时候，播报员会讲解气流流动情况。在天气预报过程中，气象专家们会根据流线图绘制的气流情况，来预测当地的天气情况。

本期，我们将学习matplotlib.pyplot.streamplot()方法相关属性的学习，let's go~

流线图概述

什么是流线图？

• 流线图通过流线和箭头的组合绘制，来表示某一时段流线的运行情况、
• 流线图上的箭头表示流向，流线上的形状表示流强度
• 流线图可分为气流图、等风速线、变高图等
• 流线图中的流线可以合并、汇合、分交，但不能交叉

流线图应用场景

流线图通常用于气象学中研究风速、气流、洋流的流向情况

流程图是风场分析的重要图表，流线的稀密度与风速大小成正比

获取流线图方法

import matplotlib.pyplot as plt 
plt.streamplot(x,y,u,v)

流线图属性

设置流线图密度

关键字：density

默认值为：1

取值类型为：浮点型或者元组

控制流线图密度，当density=1时，网格会被划分为30*30网格

对于设置每个方向上密度，可以使用元组(x,y)

设置流线宽度

关键字：linewidth

取值类型为：浮点型或者二维数组

使用二维数组，可以改变流线在网格上的线宽

阵列的形状必须要与u、v相同

设置流线颜色

关键字：color

取值可为：

• 表示颜色的英文单词：如绿色"g"
• 表示颜色单词的简称如：红色"r",黄色"y"
• RGB格式：十六进制格式如"#88c999";(r,g,b)元组形式
• 可以转入颜色列表

当使用cmap时，则需要color设置为二维数组，否则无效

设置流线缩放

关键字：norm

默认为将流线拉伸到（0，1）

仅在颜色为数组时使用

设置流线颜色系

关键字：cmap

取值形式为：颜色表_r

可取值常用的有：'Accent', 'Accent_r', 'Blues', 'Blues_r', 'BrBG', 'BrBG_r', 'BuGn', 'BuGn_r', 'BuPu', 'BuPu_r', 'CMRmap', 'CMRmap_r', 'Dark2', 'Dark2_r', 'GnBu', 'GnBu_r', 'Greens'

绘制流线图步骤

导入matplotlib.pyplot类

import matplotlib.pyplot as plt

调用numpy库arange()、random()、randint()等准备x,y,u,v数据

• x，y:一维数组/二维数组
• u,v:二维数组
• 当为二维数组，可以通过np.meshgrid(x,y),np.mgrid()创建

x = np.arange(1,10)
y = np.arange(1,10)
 
u,v = np.meshgrid(np.sin(x),np.sin(y))

调用pyplot.streamplot()绘制流线图

plt.streamplot(x,y,u,v,density=[0.5,1])

调用pyplot.show()渲染显示出流线图

plt.show()

设置linewidth、color、cmap属性绘制流线图

plt.streamplot(x,y,u,v,density=[0.5,1],color=u,cmap="Accent_r",linewidth=3)

小试牛刀

我们学习了关于绘制流线图相关属性，我们来实操一下控制流线的起点数据

• 调用np.mgrid[]定义x,y二维数据
• 调用pyplot.streamplot()方法绘制流线图
• 调用pyplot.plot()方法绘制折线图，使用marker属性标记

y,x= np.mgrid[-3:3:100j, -3:3:100j]
u = -1-x**2+y
v = 1+x-y**2
 
seed_points = np.array([[-2, -1, 0, 1, 2, -1], [-2, -1, 0, 1, 2, 2]])
plt.streamplot(x,y,u,v,density=0.6,color=u,cmap="autumn",linewidth=1,start_points=seed_points.T)
plt.plot(seed_points[0],seed_points[1],"^",color="b")
 
plt.show()

总结

本期，我们对matplotlib.pyplot提供streamplot()方法绘制流线图相关属性的学习。流线图通常使用在气象学中，研究气流变化情况。 

免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：mmqy2019@163.com进行举报，并提供相关证据，查实之后，将立刻删除涉嫌侵权内容。

猜你喜欢

• Python中怎么对栅格数据做提取和统计操作

  该教程其实源于web，我看到之后觉得很实用，于是自己又重复做了一遍，写了详细的注释分享给大家，希望对大家的研究有帮助，本文讲述了栅格的分区统计，批量提取，深化理解遍历循环等内容

• python中sys模块怎样使用?一文带你看懂

  这篇文章给大家分享的是python中sys模块的使用，对大家学习和理解python有一定的帮助，因此分享给大家做个参考，文中sys模块的使用示例介绍的非常详细，感兴趣的朋友接下来一起跟随小编看看吧。

• Pytorch搭建ShuffleNetv2怎么实现，方法是什么

  这篇文章主要介绍了Python深度学习之使用Pytorch搭建ShuffleNetv2,文中有非常详细的图文示例,对正在学习python的小伙伴们有非常好的帮助,需要的朋友可以参考下

• Python内置函数、自定义函数及main函数的用法是什么

  大家好，本篇文章主要讲的是Python的函数使用详解，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下，方便下次浏览

• Python爬虫网页元素定位什么时候用，怎样用

  这篇文章主要介绍了Python爬虫网页元素定位术，文章通过Beautiful Soup模块展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下