Найти и изменить дублированные строки в один ряд с помощью панд

Question

Если у меня есть следующий фрейм данных:

import pandas as pd
df = pd.DataFrame({"id": ['01', '02', '03', '04'], "A":["foo", "foo", "foo", "bar"], "B":[0,1,1,1], "C":["A","A","B","A"]})

   id    A  B  C
0  01  foo  0  A
1  02  foo  1  A
2  03  foo  1  B
3  04  bar  1  A

Я хочу поместить все строки в одну строку, если их значения в столбце A совпадают, как это можно сделать с пандами?спасибо.

   id   A   B  C   id   A   B   C   id    A   B   C
0  01  foo  0  A   02  foo  1   A   03   foo  1   B
1  04  bar  1  A

Answer 1

Я думаю, что следующие работы:

new_df = (df.groupby('A',as_index=False,sort=False)
          .apply(lambda x: pd.Series(x.values.flatten()))
          .unstack())    

new_df.columns = pd.np.tile(df.columns,len(new_df.columns)//len(df.columns))

>>> new_df
   id    A  B  C    id     A     B     C    id     A     B     C
0  01  foo  0  A    02   foo     1     A    03   foo     1     B
1  04  bar  1  A  None  None  None  None  None  None  None  None