Показать всю группу по столбцам

Question

'''
groupby row, concat list
'''
d = {'col1': [33, 33, 33, 34, 34, 34], 'col2': ["hello", "hello1", "hello2", "hello3", "hello4", "hello5"],
     'col3': [1, 2, 3, 4, 5, 6]}
df = pd.DataFrame(data=d)

dfQ = df.groupby('col1')['col2'].apply(list).reset_index()
print(dfQ)

Приведенный выше код дает мне col1 и col2, как отобразить результат col3 с col1 и col2?

Answer 1

Вы можете использовать agg с функцией lambda для вывода списка ваших столбцов.

dfQ = df.groupby('col1').agg(lambda x: list(x)).reset_index()

Answer 2

IIUC вы можете использовать groupby.agg

df1 = df.groupby('col1', as_index=False).agg(list)

print (df1)

   col1                  col2           col3
0   33  [hello, hello1, hello2]     [1, 2, 3]
1   34  [hello3, hello4, hello5]    [4, 5, 6]