Разделите строки на основе значения

Question

У меня есть DataFrame, который выглядит следующим образом

id  question    feedback    answer  correct False_Answer01  False_Answer02  False_Answer03
1       q1      f1          a1-1        1          NaN           NaN            NaN
1       q1      f1          a1-2        0          NaN           NaN            NaN
1       q1      f1          a1-3        0          NaN           NaN            NaN
1       q1      f1          a1-4        0          NaN           NaN            NaN
2       q2      f2          a2-1        1          NaN           NaN            NaN

И я хотел бы вставить все False Answers в False Answer столбцы - Ложный ответ - это ответ, где correct == 0. Поэтому я хочу что-то вроде этого

id  question    feedback    answer  False_Answer01  False_Answer02  False_Answer03
1       q1      f1          a1-1         a1-2           a1-3             a1-4  

Как я мог это сделать?

Answer 1

Это больше похоже на pivot проблему

df1=df.query('correct==1').copy()
df2=df.query('correct==0').copy()
df2['key']=df2.groupby(['id','question','feedback']).cumcount()+1
df2=df2.pivot_table(index=['id','question','feedback'],columns='key',values='answer',aggfunc='sum').add_prefix('False_Answer')
df=df1.dropna(1).merge(df2.reset_index(),how='left')
df
Out[388]: 
   id question feedback      ...       False_Answer1  False_Answer2  False_Answer3
0   1       q1       f1      ...                a1-2           a1-3           a1-4
1   1       q2       f2      ...                 NaN            NaN            NaN
[2 rows x 9 columns]