Python数据分析库pandas ------ merge、concatenation 、pd.concat、combine_first、stack、unstack(0)、pivot、drop;合并、拼接、组合、轴向旋转、长格式转宽格式、删除
对于合并操作,熟悉SQL的读者可以将其理解为JOIN操作,它使用一个或多个键把多行数据 结合在一起.
事实上,跟关系型数据库打交道的开发人员通常使用SQL的JOIN查询,用几个表共有的引用 值(键)从不同
的表获取数据。以这些键为基础,我们能够获取到列表形式的新数据,这些数据是对几个表中的数据进行组合
得到的。pandas库中这类操作叫作合并,执行合并操作的函数为 merge().
1 import pandas as pd
2 import numpy as np
3
4 frame1 = pd.DataFrame({
5 'color': ['white', 'red', 'red', 'black', 'green'],
6 'id':['ball','pencil','pen','mug','ashtray'],
7 'brand':['OMG','ABC','ABC','POD','POD']
8 })
9
10 frame2 = pd.DataFrame({
11 'id':['pencil','pencil','ball','pen'],
12 'brand':['OMG','POD','ABC','POD']
13 })
14
15 print(pd.merge(frame1,frame2,on='id')) # frame1,frame2的位置没有影响
16 print(pd.merge(frame2,frame1,on='id'))
17 print(pd.merge(frame2,frame1))
18 输出:
19 color id brand_x brand_y
20 0 white ball OMG ABC
21 1 red pencil ABC OMG
22 2 red pencil ABC POD
23 3 red pen ABC POD
24 id brand_x color brand_y
25 0 pencil OMG red ABC
26 1 pencil POD red ABC
27 2 ball ABC white OMG
28 3 pen POD red ABC
29 Empty DataFrame
30 Columns: [id, brand, color]
31 Index: []
没有指定按id合并的时候,合并为空,因为id,和brand两个值不能完全一致,如pencil对应的brand值,两个
表都没有一样的所以该字段就没有了。
假如两个DataFrame基准列的名称不一致,该怎样进行合并呢?为 了解决这个问题,你可以用
left_on和right_on选项指定第一个和第二个DataFrame的基准列。
1 frame2_test = frame2
2 frame2_test.columns = ['sid', 'brand'] # 这一步也会将frame2的列名改变,对比浅复制
3 print(frame1)
4 print(frame2_test)
5 print("-----*-----\n", pd.merge(frame1,frame2_test,left_on='id',right_on='sid'))
6 输出:
7 color id brand
8 0 white ball OMG
9 1 red pencil ABC
10 2 red pen ABC
11 3 black mug POD
12 4 green ashtray POD
13 sid brand
14 0 pencil OMG
15 1 pencil POD
16 2 ball ABC
17 3 pen POD
18 -----*-----
19 color id brand_x sid brand_y
20 0 white ball OMG ball ABC
21 1 red pencil ABC pencil OMG
22 2 red pencil ABC pencil POD
23 3 red pen ABC pen POD
merge()函数默认执行的是内连接操作;上述结果中的键是由交叉操作(intersection)得到的。
其他选项有左连接、右连接和外连接。外连接把所有的键整合到一起,其效果相当于左连接 和右
连接的效果之和。连接类型用how选项指定。
1 frame4 = frame1
2 print(frame2, '\n-----*frame2*-----\n') # 前面改变列名的操作已经注释掉
3 print(frame4, '\n-----*frame4*-----\n')
4 print(pd.merge(frame4,frame2,on='id'), '\n-----*内连接*-----\n')
5 print(pd.merge(frame4,frame2,on='id',how='outer'), '\n-----*外连接*-----\n')
6 print(pd.merge(frame4,frame2,on='id',how='right'), '\n-----*右连接*-----\n')
7 输出:
8 id brand
9 0 pencil OMG
10 1 pencil POD
11 2 ball ABC
12 3 pen POD
13 -----*frame2*-----
14
15 color id brand
16 0 white ball OMG
17 1 red pencil ABC
18 2 red pen ABC
19 3 black mug POD
20 4 green ashtray POD
21 -----*frame4*-----
22
23 color id brand_x brand_y
24 0 white ball OMG ABC
25 1 red pencil ABC OMG
26 2 red pencil ABC POD
27 3 red pen ABC POD
28 -----*内连接*-----
29
30 color id brand_x brand_y
31 0 white ball OMG ABC
32 1 red pencil ABC OMG
33 2 red pencil ABC POD
34 3 red pen ABC POD
35 4 black mug POD NaN
36 5 green ashtray POD NaN
37 -----*外连接*-----
38
39 color id brand_x brand_y
40 0 white ball OMG ABC
41 1 red pencil ABC OMG
42 2 red pencil ABC POD
43 3 red pen ABC POD
44 -----*右连接*-----
指定按多个列合并:
1 print(pd.merge(frame4,frame2,on=['id','brand'],how='outer'))
2 输出:
3 color id brand
4 0 white ball OMG
5 1 red pencil ABC
6 2 red pen ABC
7 3 black mug POD
8 4 green ashtray POD
9 5 NaN pencil OMG
10 6 NaN pencil POD
11 7 NaN ball ABC
12 8 NaN pen POD
这个地方如果是内连接就会是空。
根据索引合并
1 print(frame4, '\n-----*frame4*-----\n')
2 print(frame2, '\n-----*frame2*-----\n')
3 print(pd.merge(frame4,frame2,right_index=True,left_index=True))
4 # 也可以使用jion合并
5 frame2.columns = ['id2', 'brand2']
6 print(frame4.join(frame2))
7 输入:
8 color id brand
9 0 white ball OMG
10 1 red pencil ABC
11 2 red pen ABC
12 3 black mug POD
13 4 green ashtray POD
14 -----*frame4*-----
15
16 id brand
17 0 pencil OMG
18 1 pencil POD
19 2 ball ABC
20 3 pen POD
21 -----*frame2*-----
22
23 color id_x brand_x id_y brand_y
24 0 white ball OMG pencil OMG
25 1 red pencil ABC pencil POD
26 2 red pen ABC ball ABC
27 3 black mug POD pen POD
28 color id brand id2 brand2
29 0 white ball OMG pencil OMG
30 1 red pencil ABC pencil POD
31 2 red pen ABC ball ABC
32 3 black mug POD pen POD
33 4 green ashtray POD NaN NaN
两者之间的差异是非常明显的,前者不一致的就舍去,后者添加NaN补全。
拼接操作
另外一种数据整合操作叫作拼接(concatenation)。NumPy的concatenate()函数就是用于数 组的拼接操作。
1 array1 = np.arange(9).reshape((3,3))
2 array2 = np.arange(9).reshape((3,3))+6
3 # axis=0表示按行拼接,就是将array2放在array1下面
4 print(np.concatenate([array1,array2],axis=0))
5 print(np.concatenate([array1,array2],axis=1))
6 输出:
7 [[ 0 1 2]
8 [ 3 4 5]
9 [ 6 7 8]
10 [ 6 7 8]
11 [ 9 10 11]
12 [12 13 14]]
13 [[ 0 1 2 6 7 8]
14 [ 3 4 5 9 10 11]
15 [ 6 7 8 12 13 14]]
pandas库以及它的Series和DataFrame等数据结构实现了带编号的轴,它可以进一步扩展数组拼接功能。
pandas的concat()函数实现了按轴拼接的功能。
1 ser1 = pd.Series(np.random.rand(4),index=[1,2,3,4])
2 ser2 = pd.Series(np.random.rand(4),index=[5,6,7,8])
3 # pd.concat([ser1,ser2]), concat()函数默认按照axis=0这条轴拼接数据,返回Series对象。如果指定axis=l,返回结果将是DataFrame对象。
4 print(pd.concat([ser1,ser2]))
5 print(pd.concat([ser1,ser2], axis=1))
6 # 赋予新对象以等级索引
7 print(pd.concat([ser1,ser2],keys=[1,2])) #等级索引
8 frame5 = pd.DataFrame(np.random.rand(9).reshape(3,3), index=[1,2,3],columns=['A','B','C'])
9 frame2 = pd.DataFrame(np.random.rand(9).reshape(3,3), index=[4,5,6],columns=['A','B','C'])
10 print(pd.concat([frame5,frame2]))
11 print(pd.concat([frame5,frame2],axis=1))
12 输出:
13 1 0.392044
14 2 0.800669
15 3 0.899750
16 4 0.853225
17 5 0.545121
18 6 0.979369
19 7 0.864454
20 8 0.338446
21 dtype: float64
22 0 1
23 1 0.392044 NaN
24 2 0.800669 NaN
25 3 0.899750 NaN
26 4 0.853225 NaN
27 5 NaN 0.545121
28 6 NaN 0.979369
29 7 NaN 0.864454
30 8 NaN 0.338446
31 1 1 0.392044
32 2 0.800669
33 3 0.899750
34 4 0.853225
35 2 5 0.545121
36 6 0.979369
37 7 0.864454
38 8 0.338446
39 dtype: float64
40 A B C
41 1 0.781247 0.927653 0.765253
42 2 0.395887 0.625049 0.509451
43 3 0.566538 0.869041 0.862552
44 4 0.877117 0.226862 0.205766
45 5 0.489178 0.054522 0.497122
46 6 0.691023 0.076475 0.965215
47 A B C A B C
48 1 0.781247 0.927653 0.765253 NaN NaN NaN
49 2 0.395887 0.625049 0.509451 NaN NaN NaN
50 3 0.566538 0.869041 0.862552 NaN NaN NaN
51 4 NaN NaN NaN 0.877117 0.226862 0.205766
52 5 NaN NaN NaN 0.489178 0.054522 0.497122
53 6 NaN NaN NaN 0.691023 0.076475 0.965215
组合操作
还有另外一种情况,我们无法通过合并或拼接方法组合数据。例如,两个数据集的索引完全或部分重合。
1 ser1 = pd.Series(np.random.rand(5),index=[1,2,3,4,5])
2 ser2 = pd.Series(np.random.rand(4),index=[2,4,5,6])
3 # ser1组合上ser2
4 print(ser1.combine_first(ser2))
5 # ser2组合上ser1
6 print(ser2.combine_first(ser1))
7 print(ser1[:3].combine_first(ser2[:3]))
8 输出:
9 1 0.174250
10 2 0.208664
11 3 0.784141
12 4 0.861739
13 5 0.373359
14 6 0.332396
15 dtype: float64
16 1 0.174250
17 2 0.061261
18 3 0.784141
19 4 0.889045
20 5 0.232429
21 6 0.332396
22 dtype: float64
23 1 0.174250
24 2 0.208664
25 3 0.784141
26 4 0.889045
27 5 0.232429
28 dtype: float64
轴向旋转操作
1 frame5 = pd.DataFrame(np.arange(9).reshape(3,3),index=['white','black','red'],columns=['ball','pen','pencil'])
2 print(frame5)
3 frame6 = frame5.stack() # 列变行,变成层级的样式,码整齐
4 print(frame6)
5 print(frame6.unstack())
6 print(frame6.unstack(0))
7 输出:
8 ball pen pencil
9 white 0 1 2
10 black 3 4 5
11 red 6 7 8
12 white ball 0
13 pen 1
14 pencil 2
15 black ball 3
16 pen 4
17 pencil 5
18 red ball 6
19 pen 7
20 pencil 8
21 dtype: int32
22 ball pen pencil
23 white 0 1 2
24 black 3 4 5
25 red 6 7 8
26 white black red
27 ball 0 3 6
28 pen 1 4 7
29 pencil 2 5 8
长格式转宽格式
1 longframe = pd.DataFrame({
2 'color':['white','white','white','red','red','red','black','black','black'],
3 'item':['ball','pen','mug','ball','pen','mug','ball','pen','mug'],
4 'value': np.random.rand(9)
5 })
6 print(longframe)
7 输出:
8 color item value
9 0 white ball 0.993608
10 1 white pen 0.522115
11 2 white mug 0.977019
12 3 red ball 0.999250
13 4 red pen 0.051223
14 5 red mug 0.345762
15 6 black ball 0.539147
16 7 black pen 0.054598
17 8 black mug 0.420309
这种记录数据的模式有几个缺点。例如其中一个缺点是,因为一些字段具有多样性和 重复性特点,所以选
取列作为键时,这种格式的数据可读性较差,尤其是无法完全理解基准列和 其他列之间的关系。除了长格式,还
有一种把数据调整为表格形式的宽格式。这种模式可读性强,也易于连接其他表,且占用空间较少。因此一般而
言,用它存储数据效率更高,虽然它的可操作性差,这一点 尤其体现在填充数据时。如要选择一列或几列作为主
键,所要遵循的规则是其中的元素必须是唯一的。讲到格式转换,pandas提供了能够把长格式DataFrame转换为
宽格式的pivot()函数,它以用 作键的一列或多列作为参数。接着上面的例子,选择color列作为主键,item列作为
第二主键,而它们所对应的元素则作 为DataFrame的新列。
1 wideframe = longframe.pivot('color','item') 2 print(wideframe) 3 # 这种格式的DataFrame对象更加紧凑,它里面的数据可读性也更强。 4 输出: 5 value 6 item ball mug pen 7 color 8 black 0.549983 0.157802 0.987871 9 red 0.685751 0.038634 0.251137 10 white 0.263787 0.291939 0.787407
删除操作
1 frame7 = pd.DataFrame(np.arange(9).reshape(3,3),index=['white','black','red'],columns=['ball','pen','pencil'])
2 print(frame7)
3 del frame7['ball'] # 删除列
4 print(frame7)
5 frame7.drop('white') # 删除行
6 print(frame7)
7 frame7.drop('pen',axis=1) # 删除列
8 输出:
9 ball pen pencil
10 white 0 1 2
11 black 3 4 5
12 red 6 7 8
13 pen pencil
14 white 1 2
15 black 4 5
16 red 7 8
17 pen pencil
18 white 1 2
19 black 4 5
20 red 7 8