利用 Python 优雅地可视化数据

最近看《机器学习系统设计》...前两章。学到了一些用 Matplotlib 进行数据可视化的方法。在这里整理一下。

声明:由于本文的代码大部分是参考书中的例子,所以不提供完整代码,只提供示例片段,也就是只能看出某一部分用法,感兴趣的需要在自己的数据上学习测试。

最开始,当然还是要导入我们需要的包:

# -*- coding=utf-8 -*-
from matplotlib import pyplot as plt
from sklearn.datasets import load_iris
import numpy as np
import itertools

1. 画散点图

画散点图用 plt.scatter(x,y)。画连续曲线在下一个例子中可以看到,用到了 plt.plot(x,y)

plt.xticks(loc,label) 可以自定义 x 轴刻度的显示,第一个参数表示的是第二个参数 label 显示的位置 loc。

plt.autoscale(tight=True) 可以自动调整图像显示的最佳化比例 。

plt.scatter(x,y)
plt.title("Web traffic")
plt.xlabel("Time")
plt.ylabel("Hits/hour")
plt.xticks([w*7*24 for w in range(10)],['week %i' %w for w in range(10)])
plt.autoscale(tight=True)
plt.grid()
##plt.show()

画出散点图如下:

2. 多项式拟合并画出拟合曲线

## 多项式拟合
fp2 = np.polyfit(x,y,3)
f2 = np.poly1d(fp2)

fx = np.linspace(0,x[-1],1000)
plt.plot(fx,f2(fx),linewidth=4,color='g')
## f2.order: 函数的阶数
plt.legend(["d=%i" % f2.order],loc="upper right")
plt.show()

效果图:

3. 画多个子图

这里用到的是 sklearn 的 iris_dataset(鸢尾花数据集)。

此数据集包含四列,分别是鸢尾花的四个特征:

  • sepal length (cm)——花萼长度
  • sepal width (cm)——花萼宽度
  • petal length (cm)——花瓣长度
  • petal width (cm)——花瓣宽度

这里首先对数据进行一定的处理,主要就是对特征名称进行两两排列组合,然后任两个特征一个一个做 x 轴另一个做 y 轴进行画图。

# -*- coding=utf-8 -*-
from matplotlib import pyplot as plt
from sklearn.datasets import load_iris
import numpy as np
import itertools

data = load_iris()
#print(data.data)
#print(data.feature_names)
#print(data.target)

features = data['data']
feature_names = data['feature_names']
target = data['target']
labels = data['target_names'][data['target']]

print(data.data)
print(data.feature_names)

这里有一个排列组合参考代码,最后是取出了两两组合的情况。

排列组合的结果是 feature_names_2 包含了排列组合的所有情况,它的每一个元素包含了一个排列组合的所有情况,比如第一个元素包含了所有单个元素排列组合的情况,第二个元素包含了所有的两两组合的情况......所以这里取出了第二个元素,也就是所有的两两组合的情况

feature_names_2 = []
#排列组合
for i in range(1,len(feature_names)+1):
    iter = itertools.combinations(feature_names,i)
    feature_names_2.append(list(iter))
    
print(len(feature_names_2[1]))
for i in feature_names_2[1]:
    print(i)

下面是在 for 循环里画多个子图的方法。对我来说,这里需要学习的有不少。比如

  • for i,k in enumerate(feature_names_2[1]): 这一句老是记不住。
  • 比如从列表中取出某元素所在的索引的方法:index1 = feature_names.index(k[0]),也即 index = list.index(element) 的形式。
  • 比如 for 循环中画子图的方法:plt.subplot(2,3,1+i)
  • 比如 for 循环的下面这用法:for t,marker,c in zip(range(3),">ox","rgb"):

plt.figure(1)
for i,k in enumerate(feature_names_2[1]):
    index1 = feature_names.index(k[0])
    index2 = feature_names.index(k[1])
    plt.subplot(2,3,1+i)
    for t,marker,c in zip(range(3),">ox","rgb"):        
        plt.scatter(features[target==t,index1],features[target==t,index2],marker=marker,c=c)
        plt.xlabel(k[0])
        plt.ylabel(k[1])
        plt.xticks([])
        plt.yticks([])
        plt.autoscale()
        plt.tight_layout()      
plt.show()

这里的可视化效果如下:

4. 画水平线和垂直线

比如在上面最后一幅图中,找到了一种方法可以把三种鸢尾花分出来,这是我们需要画出模型(一条直线)。这个时候怎么画呢?

下面需要注意的就是 plt.vlines(x,y_min,y_max)plt.hlines(y,x_min,x_max) 的用法。

plt.figure(2)
for t,marker,c in zip(range(3),">ox","rgb"):        
    plt.scatter(features[target==t,3],features[target==t,2],marker=marker,c=c)
    plt.xlabel(feature_names[3])
    plt.ylabel(feature_names[2])
    # plt.xticks([])
    # plt.yticks([])
    plt.autoscale()    
plt.vlines(1.6, 0, 8, colors = "c",linewidth=4,linestyles = "dashed")
plt.hlines(2.5, 0, 2.5, colors = "y",linewidth=4,linestyles = "dashed")
plt.show()  

此时可视化效果如下:

5. 动态画图

plt.ion() 打开交互模式。plt.show() 不再阻塞程序运行。

注意 plt.axis() 的用法。

plt.axis([0, 100, 0, 1])
plt.ion()

for i in range(100):
    y = np.random.random()
    plt.autoscale()
    plt.scatter(i, y)
    plt.pause(0.01)

可视化效果:

文章首发公众号: CVPy

原创声明,本文系作者授权云+社区发表,未经许可,不得转载。

如有侵权,请联系 yunjia_community@tencent.com 删除。

编辑于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏Pulsar-V

原-图像处理基础(四)DCT变换

Imread()用于读取图片文件中的数据 figure() 控制画图的窗口 imshow() 显示图像 rgb2gray() 把真彩图像转变为灰度图像 d...

3146
来自专栏数据小魔方

创意滑珠图!

今天要给大家分享的是一种非常有趣的滑珠图! ▽ 本文要讲解的滑珠图做法,稍微有点复杂。不过这种滑珠图在数据对比展示中,效果奇佳。小魔方参考多处教程和资料,终于还...

2614
来自专栏数据结构与算法

06:图像相似度

06:图像相似度 总时间限制: 1000ms 内存限制: 65536kB描述 给出两幅相同大小的黑白图像(用0-1矩阵)表示,求它们的相似度。 说明:若两幅图...

3675
来自专栏数据小魔方

超强脑洞第三弹之——ggplot构造瀑布图

对,就是瀑布图,你没看错。而且是使用ggplot现有图层叠加构造,并没有用任何ggplot的外挂插件。 作图理念是在数据源的构造上,方法与《Excel图表之道》...

2683
来自专栏Flutter入门

Android OpenGL ES(八) - 简单实现绿幕抠图

这里的关键是,判断颜色的范围。这里简单的认定 g>140.0 && r<128.0 && b<128.0 时为绿色。当是绿色的时候,就将其颜色换成白色。同时al...

1132
来自专栏数据小魔方

数据标签太长了,怎么办……

今天给大家讲解在图表中长数据标签的特殊处理方法! ▽ 如果你的图表要求必须添加数据标签的话 最大的困惑就是对于哪些特别长的数据标签 加上之后图表是这样的 ? 看...

3258
来自专栏逍遥剑客的游戏开发

Direct3D学习(五):基于高度图的地形生成

17910
来自专栏WD学习记录

2016-7-3 webGL

ctx.fillRect(120,10,150,150);前两个为x、y坐标后边为长和宽

682
来自专栏Python中文社区

Python用Pillow(PIL)进行简单的图像操作

專 欄 ❈ sunhaiyu,Python中文社区专栏作者 专栏地址: http://www.jianshu.com/u/4943cb2c6ea4 ❈ Pyt...

29810
来自专栏章鱼的慢慢技术路

使用Photoshop画一个圆锥体

1975

扫码关注云+社区