python - 删除重复数据帧索引并合并列

Question

我有以下内容:

 t    len   id  accession
 t1   150   A   X1
 t1   150   B   X2
 t1   150   C   X3
 t2   200   E   X4
 t2   200   F   X5
 t2   200   G   X6

我想要以下内容:

 t    len   id      accession
 t1   150   A,B,C   X1,X2,X3 
 t2   200   E,F,G   X4, X5,X6

有没有办法像 pandas.DataFrame.pivot 中描述的那样旋转数据框？ , 但同时将列条目合并到一个列中？

Answer 1

我提出的解决方案结合了 groupby 和 agg。让我们从数据开始:

import pandas as pd

data = pd.DataFrame({
    "t": ["t1", "t1", "t1", "t2", "t2", "t2"],
    "len": [150, 150, 150, 200, 200, 200],
    "id": ["A", "B", "C", "E", "F", "G"],
    "accession": ["X1", "X2", "X3", "X4", "X5", "X6"]})

解决方案

data.groupby(["t", "len"]).agg({"id": lambda x: ",".join(x), 
                                "accession": lambda x: ",".join(x)}).reset_index()

输出

    t  len     id accession
0  t1  150  A,B,C  X1,X2,X3
1  t2  200  E,F,G  X4,X5,X6