python数据分析:详解数据框的合并
不知道大家有没有分析过Kaggle的数据或者参加过Kaggle的挑战,里面的数据都分布在几个不同的数据集中,合并数据在所难免。今天给大家详细总结一波pandas中数据框合并和连接的方法。建议收藏,哈哈。
生成练习用的数据框
首先我们先用如下代码生成3个数据框,作为演示讲解之用。
df1 =pd.DataFrame({
'id': ['1', '2', '3', '4', '5'],
'Feature1': ['A', 'C', 'E', 'G', 'I'],
'Feature2': ['B', 'D', 'F', 'H', 'J']})
df2 =pd.DataFrame({
'id': ['1', '2', '6', '7', '8'],
'Feature1': ['K', 'M', 'O', 'Q', 'S'],
'Feature2': ['L', 'N', 'P', 'R', 'T']})
df3 = pd.DataFrame({
'id': ['1', '2', '3', '4', '5', '7', '8', '9', '10', '11'],
'Feature3': [12, 13, 14, 15, 16, 17, 15, 12, 13, 23]})
得到3个数据框,如下:
连接数据框Concatenate DataFrames
如果只是简单的连接两个数据框的行的话,直接调用concat()方法即可
df_row = pd.concat([df1, df2])
df_row
可以看到,数据框df1和df2就以行连接在一起了,但是你注意到没,此时数据框的索引有问题,我们想要的索引应该自动给我们填好才对,这个时候我们需要将ignore_index参数设置为True。
df_row = pd.concat([df1, df2], ignore_index=True)
这个时候索引就正确了。
有时候,你虽然连接了2个数据框,但是你还是想知道新数据框的数据分别来自哪个数据框,这个时候调用keys参数就可以,例如,我们在上面的例子中将来自df1的数据和df2的数据分别加上标签x,y。
df_keys = pd.concat([df1,df2], keys=['x', 'y'])
df_keys
可以看到,新数据框中哪些数据来自于哪个数据框一目了然。
同时,给新的数据框加上标签也有助于我们在新数据中切片出旧的数据框,例如
df_keys.loc['y']
通过以上代码,很容易在新数据框中切片出了df2。
在连接数据框时还有一个技巧,就是将数据框以字典值的形式传递给 concat(),可以在连接的新数据框中自动加上标签,这个标签就是字典的键。看下例:
df_piece = pd.concat({'第一个框':df1,'第二个框':df2})
df_piece
有没有很好使?
concat()函数也可以横向连接数据框,只需要将axis设置为1即可
df_col = pd.concat([df1,df2], axis=1)
df_col
可以看到这个就是横向连接。
合并数据框Merge DataFrames
我们看一下如何将df_row(由df1和df2连接后得到,见之前的例子)和df3以共同的id为基础合并起来。
df_merge_col = pd.merge(df_row, df3, on='id')
df_merge_col
此时,我们注意到id等于1的行出现了2次,是因为在df_row中是有2个id等于1的,而在df3中id等于1只有一次,所以合并后的feature3都是12。
还有一种情况是你要合并的2个数据框没有一样名字的列,需要你指定,这个时候就要用到left_on,right_on参数了。
df_merge_difkey = pd.merge(df_row, df3, left_on='id', right_on='id')
df_merge_difkey
此例中我们左右数据框都指定以id为基础进行合并。
还有一种情况是,你想给原先的数据框加一行新数据,看下例:
add_row = pd.Series(['10', 'X1', 'X2', 'X3'],
index=['id','Feature1', 'Feature2', 'Feature3'])
df_add_row = df_merge_col.append(add_row, ignore_index=True)
df_add_row
可以看到,通过append可以给一个数据框加上一个新的series。
不同的合并逻辑Outer,Inner,Right,Left与index
Outer合并会将两个数据库的所有数据都合并,相当于取原来数据库的并集形成一个新的数据库。
df_outer = pd.merge(df1, df2, on='id', how='outer')
df_outer
可以看到,这种合并方法,生成的新数据集中会有很多NaN的值,还有,需要注意的是这种方法会自动将相同列名加上后缀,而且这个后缀是可以改的,看下例:
df_suffix = pd.merge(df1, df2, left_on='id',right_on='id',how='outer',suffixes=('_left','_right'))
df_suffix
可以看到suffixes参数可以修改后缀。
Inner合并生成的新数据集中只会有原来2个数据集中都有的数据,相当于取了两个数据框的交集。
df_inner = pd.merge(df1, df2, on='id', how='inner')
df_inner
相应的,Right,和Left就是分别以第二个和第一个数据框中的变量为基准进行数据框的合并,大家也可以试试。有时候,我们会根据2个数据框的index来合并,此时,只需要将right_index,left_index两个参数设置为True即可。看下面例子:
df_index = pd.merge(df1, df2, right_index=True, left_index=True)
df_index
结论
好了,今天给大家介绍了concat() 和 merge()合并数据框的用法,希望对大家有帮助。感谢大家耐心看完。发表这些东西的主要目的就是督促自己,希望大家关注评论指出不足,一起进步。内容我都会写的很细,用到的数据集也会在原文中给出链接,你只要按照文章中的代码自己也可以做出一样的结果,一个目的就是零基础也能懂,因为自己就是什么基础没有从零学Python的,加油。
(站外链接发不了,请关注后私信回复“数据链接”获取本头条号所有使用数据)
往期精彩: