hierarchical indexing是pandas
的一項重要功能,它使你能在一個軸上擁有多個(兩個以上)索引級別。它能讓你以低維度形式處理高維度數據。
例:創建一個Series
,并用一個由列表或數組組成的列表作為索引:
In [80]: data = Series(np.random.randn(10), index=[['a','a','a','b','b','b','c','c','d','d’], [1,2,3,1,2,3,1,2,2,3]])
In [81]: data
Out[81]:
a 1 0.659838
2 -0.530732
3 0.862788
b 1 -0.678278
2 1.189240
3 0.167645
c 1 1.740861
2 0.144723
d 2 1.153850
3 0.686639
dtype: float64
這就是帶有MultiIndex
索引的Series
的格式化輸出形式;索引之間的“間隔”表示“直接使用上面的標簽”:
In [82]: data.index
Out[82]:
MultiIndex(levels=[['a', 'b', 'c', 'd'], [1, 2, 3]],
labels=[[0, 0, 0, 1, 1, 1, 2, 2, 3, 3], [0, 1, 2, 0, 1, 2, 0, 1, 1, 2]])
對于一個層次化索引對象選取子集:
In [83]: data['b']
Out[83]:
1 -0.678278
2 1.189240
3 0.167645
dtype: float64
In [84]: data['b':'d']
Out[84]:
b 1 -0.678278
2 1.189240
3 0.167645
c 1 1.740861
2 0.144723
d 2 1.153850
3 0.686639
dtype: float64
有時甚至還可以在“內層”中進行選取:
In [85]: data[:,2]
Out[85]:
a -0.530732
b 1.189240
c 0.144723
d 1.153850
dtype: float64
層次化索引在數據重塑和基于分組的操作(如透視表生成)中扮演著重要的角色。
比如說,這段數據可以通過unstack
方法被重新安排到一個DataFrame
中:
In [86]: data.unstack()
Out[86]:
1 2 3
a 0.659838 -0.530732 0.862788
b -0.678278 1.189240 0.167645
c 1.740861 0.144723 NaN
d NaN 1.153850 0.686639
unstack
的逆運算是stack
:
對于DataFrame
,每條軸都可以有分層索引:
In [88]: frame = DataFrame(np.arange(12).reshape((4,3)), index=[['a','a','b','b'], [1,2,1,2]], columns=[['Ohio','Ohio','Colorado'],['Green','Red','Green']])
In [89]: frame
Out[89]:
Ohio Colorado
Green Red Green
a 1 0 1 2
2 3 4 5
b 1 6 7 8
2 9 10 11
各層都可以有名字(可以是字符串,也可以是別的Python對象)。如果指定了名稱,它們就會顯示在控制臺輸出中(注意區別索引名稱跟軸標簽!):
In [90]: frame.index.names = ['key1', 'key2']
In [93]: frame.columns.names = ['state', 'color']
In [94]: frame
Out[94]:
state Ohio Colorado
color Green Red Green
key1 key2
a 1 0 1 2
2 3 4 5
b 1 6 7 8
2 9 10 11
由于有了分部的列索引,因此可以輕松選取列分組:
In [95]: frame['Ohio']
Out[95]:
color Green Red
key1 key2
a 1 0 1
2 3 4
b 1 6 7
2 9 10
可以單獨創建MultiIndex
然后復用:
columns = pd.MultiIndex.from_arrays([['Ohio','Ohio','Colorado'],['Green','Red','Green'] ...: ], names=['state','color'])
重新分級排序:
重新調整某條軸上各級別的順序,或根據指定級別上的值對數據進行排序。
swaplevel
接受兩個編號或名稱,并返回一個互換了級別的新對象:
In [102]: frame.swaplevel('key1', 'key2')
Out[102]:
state Ohio Colorado
color Green Red Green
key2 key1
1 a 0 1 2
2 a 3 4 5
1 b 6 7 8
2 b 9 10 11
sort_index
則根據單個級別中的值對數據進行排序(穩定的):
In [103]: frame.sortlevel(1)
/Users/suhang/anaconda3/bin/ipython:1: FutureWarning: sortlevel is deprecated, use sort_index(level= ...)
#!/Users/suhang/anaconda3/bin/python
Out[103]:
state Ohio Colorado
color Green Red Green
key1 key2
a 1 0 1 2
b 1 6 7 8
a 2 3 4 5
b 2 9 10 11
In [104]: frame.swaplevel(0,1).sort_index(0)
Out[104]:
state Ohio Colorado
color Green Red Green
key2 key1
1 a 0 1 2
b 6 7 8
2 a 3 4 5
b 9 10 11
根據級別匯總統計:
許多對DataFrame
和Series
的描述和匯總統計都有一個level
選項,它用于指定在某條軸上求和的級別。
In [105]: frame
Out[105]:
state Ohio Colorado
color Green Red Green
key1 key2
a 1 0 1 2
2 3 4 5
b 1 6 7 8
2 9 10 11
根據行或列上的級別進行求和:
In [107]: frame.sum(level='key2')
Out[107]:
state Ohio Colorado
color Green Red Green
key2
1 6 8 10
2 12 14 16
In [110]: frame.sum(level='color', axis=1)
Out[110]:
color Green Red
key1 key2
a 1 2 1
2 8 4
b 1 14 7
2 20 10
其實就是利用了pandas
的groupby
功能,稍后講。
使用DataFrame
的列:
人們經常想將DataFrame的一個或多個列當做行索引來用,或者將行索引變成DataFrame
的列:
In [111]: frame = DataFrame({'a':range(7), 'b':range(7,0,-1), 'c':['one', 'one', 'one', 'two', 'two', 'two', 'two'], 'd':[0,1,2,0,1,2,3]})
In [112]: frame
Out[112]:
a b c d
0 0 7 one 0
1 1 6 one 1
2 2 5 one 2
3 3 4 two 0
4 4 3 two 1
5 5 2 two 2
6 6 1 two 3
set_index
函數會將其一個或多個列轉換為行索引,并創建一個新的DataFrame
:
In [113]: frame2 = frame.set_index(['c','d'])
In [114]: frame2
Out[114]:
a b
c d
one 0 0 7
1 1 6
2 2 5
two 0 3 4
1 4 3
2 5 2
3 6 1
默認情況下,那些列會被移除,也可以保留下來,drop=False
:
In [115]: frame.set_index(['c','d'], drop=False)
Out[115]:
a b c d
c d
one 0 0 7 one 0
1 1 6 one 1
2 2 5 one 2
two 0 3 4 two 0
1 4 3 two 1
2 5 2 two 2
3 6 1 two 3
reset_index
的功能跟set_index
相反,層次化索引的級別會被移到列里面:
In [116]: frame2.reset_index()
Out[116]:
c d a b
0 one 0 0 7
1 one 1 1 6
2 one 2 2 5
3 two 0 3 4
4 two 1 4 3
5 two 2 5 2
6 two 3 6 1