python - 将 Pandas 数据框的行放入列表形式

Question

我有一个表格的 Pandas 数据框

    T1  T2
0   A   B
1   C   D
2   B   C
3   D   E
4   F   A

我想生成另一个 Pandas 数据框，其中列出了 T1 和 T2 中的每个唯一项目都有自己的行，并且有一列包含该唯一项目的名称，一列包含与它共享一行的项目列表原始数据框。例如，在这种情况下，我将寻找以下形式的内容:

    Name    List
0   A   [B, F]
1   B   [A, C]
2   C   [D, B]
3   D   [C, E]
4   E   [D]
5   F   [A]

有人可以建议一个适当的 pandonic(比如 pythonic 但对于 Pandas :))来做到这一点？提前致谢!

Answer 1

IIUC，交换列并使用 pandas.DataFrame.columns :

df2 = df.copy()
df2.columns = df.columns[::-1]
new_df = pd.concat([df, df2])
new_df.groupby("T1")["T2"].apply(list).reset_index()

输出:

  T1      T2
0  A  [B, F]
1  B  [C, A]
2  C  [D, B]
3  D  [E, C]
4  E     [D]
5  F     [A]