今天我们来聊一下​​Pandas​​当中的数据集中带有多重索引的数据分析实战

通常我们接触比较多的是单层索引,而多级索引也就意味着数据集当中的行索引有多个层级,具体的如下图所示

Pandas 多层级索引 Python 数据处理案例指南_python

导入数据

我们先导入数据与​​pandas​​模块

import pandas as pd

## 导入数据集
df = pd.read_csv('dataset.csv')
df.head()

output

Pandas 多层级索引 Python 数据处理案例指南_数据集_02

该数据集描述的是​英国部分城市在2019年7月1日至7月4日期间的全天天气状况​,我们先来看一下当前的数据集的行索引有哪些?代码如下

df.index.names

output

FrozenList(['City', 'Date'])

数据集当中​​City​​​、​​Date​​​,这里的​​City​​我们可以当作是​第一层级索引​,而​​Date​​则是​第二层级索引​。

我们也可以通过调用​​sort_index()​​方法来按照数据集的行索引来进行排序,代码如下

df_1 = df.sort_index()
df_1

output

Pandas 多层级索引 Python 数据处理案例指南_数据分析_03

要是我们想将这个多层索引去除掉,就调用​​reset_index()​​方法,代码如下

df.reset_index()

下面我们就开始针对多层索引来对数据集进行一些分析的实战吧

第一层级的数据筛选

在​​pandas​​​当中数据筛选的方法,一般我们是调用​​loc​​​以及​​iloc​​方法,同样地,在多层级索引的数据集当中数据的筛选也是调用该两种方法,例如筛选出伦敦白天的天气状况如何,代码如下

df_1.loc['London' , 'Day']

output

Pandas 多层级索引 Python 数据处理案例指南_数据集_04

要是我们想针对所有的行,就可以这么来做

df_1.loc[:, 'Day']

output

Pandas 多层级索引 Python 数据处理案例指南_python_05

同理针对所有的列,就可以这么来做

df_1.loc['London' , :]

output

Pandas 多层级索引 Python 数据处理案例指南_数据挖掘_06

多层级索引的数据筛选

要是我们想看伦敦2019年7月1日白天的天气状况,就可以这么来做

df.loc['London', 'Day'].loc['2019-07-01']

output

Weather               Shower
Wind SW 16 mph
Max Temperature 28
Name: 2019-07-01, dtype: object

这里我们进行了两次数据筛选的操作,先是​​df.loc['London', 'Day']​​​,然后再此的基础之上再进行​​loc['2019-07-01']​​操作,当然还有更加方便的步骤,代码如下

df.loc[('London', '2019-07-01'), 'Day']

output

Weather               Shower
Wind SW 16 mph
Max Temperature 28
Name: 2019-07-01, dtype: object

除此之外我们要是想看一下伦敦2019年7月1日和7月2日两天白天的天气情况,就可以这么来做

df.loc[ 
('London' , ['2019-07-01','2019-07-02'] ) ,
'Day'
]

output

Pandas 多层级索引 Python 数据处理案例指南_数据分析_07

在此基础之上,我们想要看天气和风速这两列,我们也可以单独摘出来,代码如下

df.loc[ 
'London' ,
('Day', ['Weather', 'Wind'])
]

output

Pandas 多层级索引 Python 数据处理案例指南_数据集_08

按照范围来筛选数据

对于第一层级的索引而言,我们同样还是调用​​loc​​方法来实现

df.loc[
'Cambridge':'Oxford',
'Day'
]

output

Pandas 多层级索引 Python 数据处理案例指南_数据分析_09

但是对于第二层级的索引,要是用同样的方式来用就会报错,

df.loc[
('London', '2019-07-01': '2019-07-03'),
'Day'
]

output

SyntaxError: invalid syntax (<ipython-input-22-176180497f92>, line 3)

正确的写法代码如下

df.loc[
('London','2019-07-01'):('London','2019-07-03'),
'Day'
]

output

Pandas 多层级索引 Python 数据处理案例指南_数据集_10

筛选出所有全部的内容

对于单层索引而言,我们通过​​:​​来筛选出所有的内容,但是在多层级的索引上面则并不适用,

# 出现语法错误
df.loc[
('London', :),
'Day'
]
# 出现语法错误
df.loc[
(: , '2019-07-04'),
'Day'
]

正确的做法如下所示

# 筛选出伦敦下面所有天数的白天天气情况
df.loc[
('London', slice(None)),
'Day'
]

output

Pandas 多层级索引 Python 数据处理案例指南_数据挖掘_11

# 筛选出2019年7月4日下所有城市的白天天气情况
df.loc[
(slice(None) , '2019-07-04'),
'Day'
]

output

Pandas 多层级索引 Python 数据处理案例指南_数据挖掘_12

当然这里还有更加简便的方法,我们通过调用​​pandas​​​当中​​IndexSlice​​函数来实现,代码如下

from pandas import IndexSlice as idx
df.loc[
idx[: , '2019-07-04'],
'Day'
]

output

Pandas 多层级索引 Python 数据处理案例指南_数据挖掘_13

又或者是

rows = idx[: , '2019-07-01']
cols = idx['Day' , ['Max Temperature','Weather']]
df.loc[rows, cols]

output

Pandas 多层级索引 Python 数据处理案例指南_数据集_14

​xs()​​方法的调用

对于多层级索引的数据集而言,调用​​xs()​​方法能够更加方便地进行数据的筛选,例如我们想要筛选出日期是2019年7月4日的所有数据,代码如下

df.xs('2019-07-04', level='Date')

output

Pandas 多层级索引 Python 数据处理案例指南_数据_15

我们需要在​​level​​参数上指定是哪个标签,例如我们想要筛选出伦敦2019年7月4日全天的天气情况,代码如下

df.xs(('London', '2019-07-04'), level=['City','Date'])

output

Pandas 多层级索引 Python 数据处理案例指南_数据分析_16

最后​​xs​​​方法可以和上面提到的​​IndexSlice​​函数联用,针对多层级的数据集来进行数据的筛选,例如我们想要筛选出2019年7月2日至7月4日,伦敦全天的天气状况,代码如下

rows= (
idx['2019-07-02':'2019-07-04'],
'London'
)
df.xs(
rows ,
level = ['Date','City']
)

output

Pandas 多层级索引 Python 数据处理案例指南_python_17

技术交流

欢迎载、收藏、有所收获点赞支持一下!

Pandas 多层级索引 Python 数据处理案例指南_数据_18

目前开通了技术交流群,群友已超过​2000人​,添加时最好的备注方式为:来源+兴趣方向,方便找到志同道合的朋友

  • 方式①、发送如下图片至微信,长按识别,后台回复:加群;
  • 方式③、微信搜索公众号:​Python学习与数据挖掘​,后台回复:加群

Pandas 多层级索引 Python 数据处理案例指南_数据_19