通过比较多列组合pandas DataFrame中的行

Question

如何从以下 DataFrame 中获得预期结果

 df
            col_1             col_2    col_3     col_4  
     0  Non-Saved    www.google.com     POST    20,567
     1               www.google.com     POST
     2             www.facebook.com      GET   
     3             www.facebook.com    OTHER
     4             www.linkedin.com      GET
     5      Saved     www.Quora.com     POST     6,337
     6                www.gmail.com     POST 
     7                www.gmail.com      GET

预期结果：

            col_1             col_2    col_3     col_4  
     0  Non-Saved    www.google.com     POST    20,567
                   www.facebook.com      GET   
                   www.linkedin.com    OTHER
     1      Saved     www.Quora.com     POST     6,337
                      www.gmail.com      GET

通过合并 col_1 和 col_3 中的空字符串，从 8 行到 2 行。 此外，将 col_2 和 col_3 中的不同值连接到一个单元格中。 任何人都可以帮助我使用用户定义的函数来做到这一点吗？

Answer 1

如果之前的解决方案有效，那么让我们试试这个：

l = lambda x: ' , '.join(x.unique())

df = df.apply(lambda x: x.str.strip()).replace('',np.nan)

print(df.groupby(df.col_1.ffill())\
  .agg({'col_2': l,'col_3': l, 'col_4':'first'})\
  .reset_index())

输出：

       col_1                                              col_2  \
0  Non-Saved  www.google.com , www.facebook.com , www.linked...   
1      Saved                      www.Quora.com , www.gmail.com   

                col_3   col_4  
0  POST , GET , OTHER  20,567  
1          POST , GET   6,337