将熊猫数据框的行放入列表形式

Question

我有一个表单的 pandas 数据框

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

我想生成另一个熊猫数据框，其中列出了 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]