Как разделить данные из одного столбца фрейма данных pandas на несколько столбцов нового фрейма данных ⇐ Python

[Anonymous]

Я хотел бы разделить данные из этого кадра данных pandas (назовем его df1):

Код: Выделить всё

YEAR   CODE   DIFF
2013   XXXX   5.50
2013   YYYY   8.50
2013   ZZZZ   6.50
2014   XXXX   4.50
2014   YYYY   2.50
2014   ZZZZ   3.50

Таким образом, я создаю новый фрейм данных (назовем его df2), который выглядит следующим образом:

Код: Выделить всё

YEAR   XXXX_DIFF   ZZZZ_DIFF
2013   5.50        6.50
2014   4.50        3.50

Думаю, я группирую по годам и разделяю результаты по одному столбцу, найденные в DIFF, на конкретные совпадения CODE. Я пробовал что-то вроде этого:

Код: Выделить всё

df2 = df1[['YEAR','CODE','DIFF']].query('CODE == "XXXX"')

И я знаю, что могу переименовывать столбцы и удалять лишние, но я не был уверен, как передать значения ZZZZ DIFF в df2?

Подробнее здесь: https://stackoverflow.com/questions/599 ... tiple-colu