python - 如何在 pandas 中添加 concat 指示符列？

Question

使用 merge() 有一个指示符参数，因此您可以识别从哪个 df 中找到原始数据。我没有看到与 concat() 类似的函数。

pd.merge(left=df1, right=df2, indicator=True)

有没有办法使用concat()创建指示器列？

Answer 1

import pandas as pd

df1 = pd.DataFrame(['a','b'], index=[0,1], columns=['letter'])
df2 = pd.DataFrame(['c','d','e'], index=[2,3,4], columns=['letter'])
result = pd.concat([df1, df2], axis=0, keys=['from_df1','from_df2']).reset_index(level=[0])

df1

  letter
0      a
1      b

df2

  letter
2      c
3      d
4      e

结果

    level_0 letter
0  from_df1      a
1  from_df1      b
2  from_df2      c
3  from_df2      d
4  from_df2      e