合并具有多个值的pandas数据帧

时间:2017-04-28 18:30:41

标签: python pandas

我有pandas数据框,如:

_data_orig = [
        [1, 3.2],
        [3, 3.9],
        [4, 1.2],
        [5, 2.2]
    ]
_columns1 = ["ID", "GPA"]

_data_new = [
    [1, "Bob"],
    [2, "Sam"],
    [3, "Jane"],
    [3, "Sanoj"]
]
_columns2 = ["ID", "Name"]


df_orig = pd.DataFrame(data=_data_orig, columns=_columns1)
df_new = pd.DataFrame(data=_data_new, columns=_columns2)

当我这样做时:

df_merge = pd.merge(df_orig, df_new, how='left')

我明白了:

    ID  GPA Name
0   1   3.2 Bob
1   3   3.9 Jane
2   3   3.9 Sanoj
3   4   1.2 NaN
4   5   2.2 NaN

您可以看到重复ID:3。我想要这种格式,以便ID:3不会从df_orig重复:

    ID  GPA Name    Name_1
0   1   3.2 Bob 
1   3   3.9 Jane    Sanoj
2   4   1.2 NaN 
4   5   2.2 NaN 

2 个答案:

答案 0 :(得分:1)

试试这个:

让我们创建以下助手DF

In [279]: x = (df_new.groupby('ID')['Name']
     ...:            .apply(';'.join)
     ...:            .str.split(';', expand=True)
     ...:            .add_prefix('Name_')
     ...:            .reset_index())
     ...:

In [280]: x
Out[280]:
   ID Name_0 Name_1
0   1    Bob   None
1   2    Sam   None
2   3   Jane  Sanoj

现在我们可以简单地将其与df_orig DF

合并
In [281]: pd.merge(df_orig, x, how='left').fillna('')
     ...:
Out[281]:
   ID  GPA Name_0 Name_1
0   1  3.2    Bob
1   3  3.9   Jane  Sanoj
2   4  1.2
3   5  2.2

答案 1 :(得分:1)

pivot groupby().cumcount与[{1}}:

一起考虑merge
df_new['IDcount'] = "Name_" + (df_new.groupby("ID").cumcount() + 1).astype(str)
df_wide = df_new.pivot(index="ID", columns="IDcount", values="Name").reset_index()

df_merge = pd.merge(df_orig, df_wide, on='ID', how='left')

#    ID  GPA Name_1 Name_2
# 0   1  3.2    Bob   None
# 1   3  3.9   Jane  Sanoj
# 2   4  1.2    NaN    NaN
# 3   5  2.2    NaN    NaN