前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Pandas绘图功能

Pandas绘图功能

作者头像
用户3577892
发布2020-11-12 10:45:29
1.7K0
发布2020-11-12 10:45:29
举报
文章被收录于专栏:数据科学CLUB数据科学CLUB

目录

  • 柱状图
  • 箱线图
  • 密度图
  • 条形图
  • 散点图
  • 折线图
  • 保存绘图
  • 总结

可视化是用来探索性数据分析最强大的工具之一。Pandas库包含基本的绘图功能,可以让你创建各种绘图。Pandas中的绘图是在matplotlib之上构建的,如果你很熟悉matplotlib你会惊奇地发现他们的绘图风格是一样的。

本案例用到的数据集是关于钻石的。

代码语言:javascript
复制
import numpy as np
import pandas as pd
import matplotlib

%matplotlib inline
代码语言:javascript
复制
diamonds = pd.read_csv("diamonds.csv")
diamonds = diamonds.drop("Unnamed: 0", axis=1)

print(diamonds.shape)        

diamonds.head(5)
代码语言:javascript
复制
(53940, 10)

输出结果显示,数据集包含53940个不同钻石的10个特征,其中有数值变量也有分类变量。

柱状图

柱状图是一个单变量图(注意区分柱状图和条形图),它将一个数值变量分组到各个数值单元中,并显示每个单元中的观察值数量。直方图是了解数值变量分布的一种有用工具。所用到的方法是df.hist()

代码语言:javascript
复制
diamonds.hist(column="carat",        # 具体列
              figsize=(8,8),         # 图片大小
              color="blue");         # 绘画颜色

从图上我们可以看到钻石重量的分布是十分倾斜的:大多数钻石大约1克拉及以下,但也有极少量极端值。 为了获得更多细节的数据,我们可以增加分箱的数量来查看更小范围内的钻石重量,通过限制x轴的宽度使整个图形在画布上显得不那么拥挤。

代码语言:javascript
复制
diamonds.hist(column="carat",        
              figsize=(8,8),         
              color="blue",          
              bins=50,               
              range= (0,3.5));       

这个直方图让我们更好地了解了分布中的一些细微差别,但我们不能确定它是否包含所有数据。将X轴限制在3.5可能会剔除一些异常值,以至于它们在原始图表中没有显示。接下来看看有没有钻石大于3.5克拉:

代码语言:javascript
复制
diamonds[diamonds["carat"] > 3.5]

哦豁,真的有9颗钻石比3.5克拉大,这些'怪种'钻石我们应该关心吗?出于数据探索的目的,我们完全可以舍弃这些点,但如果是把数据的全貌展示给别人看,我觉得有必要详细说明:范围之外还存在9个离群点。

箱线图

箱线图是另一种单变量图, 方法pd.boxplot()

代码语言:javascript
复制
diamonds.boxplot(column="carat");

箱线图的中心框代表中间50%的观察值,中心线代表中位数。 boxplot最有用的特性之一是能够生成并排的boxplots。每个分类变量都在一个不同的boxside上绘制一个分类变量。接下来将钻石价格按钻石净度分成两部分来做一个并排的方框图:

代码语言:javascript
复制
diamonds.boxplot(column="price",        
                 by= "clarity",         
                 figsize= (8,8));       

上面的箱线图很奇怪:按理说清晰度更好的钻石能卖到更高的价格,然而清晰度最高的钻石(IF)的中间价却比低净度钻石低!这是为什么呢?也许下面这个图可以给我们一些启示:

代码语言:javascript
复制
diamonds.boxplot(column="carat",        
                 by= "clarity",         
                 figsize= (8,8));       

上面的图表显示,透明度较低的钻石往往更大,透明度高的钻石更加小巧。由于尺寸重量是决定钻石价值的另一个重要因素,因此低透明度钻石的中间价较高也就不足为奇了。

密度图

密度图以连续曲线显示数值变量的分布。它类似于柱状图,但密度图能更好地显示分布的基本形状。series.plot(kind="density")

代码语言:javascript
复制
diamonds["carat"].plot(kind="density", 
                      figsize=(8,8),    
                      xlim= (0,5));     

条形图

条形图是直观显示分类变量计数的图形,df.plot(kind="bar"):

代码语言:javascript
复制
carat_table = pd.crosstab(index=diamonds["clarity"], columns="count")
carat_table
代码语言:javascript
复制
carat_table.plot(kind="bar",
                 figsize=(8,8));

可以使用二维表格创建堆积条形图。堆积条形图显示每个条形图中另一个变量的分布:

代码语言:javascript
复制
carat_table = pd.crosstab(index=diamonds["clarity"], 
                          columns=diamonds["color"])

carat_table
代码语言:javascript
复制
carat_table.plot(kind="bar", 
                 figsize=(8,8),
                 stacked=True);

分组条形图是堆叠条形图的另一种选择,设置stacked=False即可:

代码语言:javascript
复制
carat_table.plot(kind="bar", 
                 figsize=(8,8),
                 stacked=False);

散点图

散点图是双变量图,采用两个数值变量,并在x/y平面上绘制数据点。创建单个散点图使用方法df.plot(kind="scatter"):

代码语言:javascript
复制
diamonds.plot(kind="scatter",     # Create a scatterplot
              x="carat",          # Put carat on the x axis
              y="price",          # Put price on the y axis
              figsize=(10,10),
              ylim=(0,20000));

尽管上面的散点图有许多重叠点,但它仍然让我们对钻石克拉重量和价格之间的关系有了一些了解:大钻石通常更贵。

折线图

折线图通常用于绘制时间序列数据:

代码语言:javascript
复制
years = [y for y in range(1950,2016)]

readings = [(y+np.random.uniform(0,20)-1900) for y in years]

time_df = pd.DataFrame({"year":years,
                        "readings":readings})


time_df.plot(x="year",
             y="readings",
             figsize=(9,9));

保存绘图

如果要保存图片供以后使用,两步就可以轻松解决:首先用plot.get_figure(),然后用figure.savefig("filename")。图片可以保存为多种常见的文件格式,例如png、jpeg和pdf。

代码语言:javascript
复制
my_plot = time_df.plot(x="year",    
             y="readings",
             figsize=(9,9))

my_fig = my_plot.get_figure()            

my_fig.savefig("line_plot_example.png")  

总结

Python绘图生态系统有许多不同的库,大部分人可能会很难从中抉择,不知道该如何人下手。Pandas绘图函数使你能够快速地可视化和浏览数据。Pandas绘图函数并没有提供尽善尽美的所有功能,但它们通常足以完成任务。

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2020-11-06,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 数据科学CLUB 微信公众号,前往查看

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

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 目录
  • 柱状图
  • 箱线图
  • 密度图
  • 条形图
  • 散点图
  • 折线图
  • 保存绘图
  • 总结
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档