python - pandas:如何在一列上合并具有相同列名的多个数据框？

Question

我有 N 个数据框:

df1:
time  data
1.0   a1
2.0   b1
3.0   c1


df2:
time  data
1.0   a2
2.0   b2
3.0   c2



df3:
time  data
1.0   a3
2.0   b3
3.0   c3

我想把它们全部合并到id上，从而得到

time  data1    data2    data3
1.0   a1       a2       a3
2.0   b1       b2       b3
3.0   c1       c2       c3

我可以确保所有数据框中的所有 ID 都相同。

我如何在 pandas 中执行此操作？

Answer 1

一个想法是使用 concat对于 DataFrame 列表 - 只需要为每个 DaatFrame 创建索引 id。还为避免重复的列名称添加了 keys 参数，但它在输出中创建了 MultiIndex。所以添加了 map 和 format 来展平它:

dfs = [df1, df2, df3]
dfs = [x.set_index('id') for x in dfs]

df = pd.concat(dfs, axis=1, keys=range(1, len(dfs) + 1))
df.columns = df.columns.map('{0[1]}{0[0]}'.format)
df = df.reset_index()
print (df)
   id data1 data2 data3
0   1    a1    a2    a3
1   2    b1    b2    b3
2   3    c1    c2    c3