Pandas 通過保留一列的第一個元素和另一列的最后一個元素來合並

Question

我有一個具有不同值和 ID 的數據框，它們可以是共同的。

    df = pd.DataFrame({'A': ['chr1','chr1','chr1','chr1','chr1','chr2'],
                    'B': [700,750,800,850,900,200],
                    'C': [750,800,850,900,950,250],
                    'D':['id_1','id_1','id_1','id_1','id_1','id_2']})

我想要做的是保持 B 的最低元素，對於相同的值 A 和 D 保持較高的 C

輸出應該是：

    A    B    C    D
0  chr1 700  950   id_1
1  chr2 200  250   id_2

我試着用

groupby('D').agg(['first', 'last'])

但這不是我想要的......

Answer 1

將GroupBy.agg與按列名稱和聚合函數的字典一起使用：

df1 = (df.groupby('D', as_index=False)
         .agg({'A':'first', 'B':'first', 'C':'last'})
         [['A','B','C','D']])
print (df1)
      A    B    C     D
0  chr1  700  950  id_1
1  chr2  200  250  id_2

Answer 2

使用dict在agg傳遞名稱和函數

df.groupby(['A','D'],as_index=False).agg({'B':'first','C':'last'}).reindex(columns=df.columns)
      A    B    C     D
0  chr1  700  950  id_1
1  chr2  200  250  id_2