Коллапсирующие ряды в Колонны Панд - PullRequest
Новая
       18

Коллапсирующие ряды в Колонны Панд

2 голосов
/ 16 марта 2019

Я пытаюсь свернуть фрейм данных из нескольких строк в pandas. Допустим, что фрейм данных выглядит следующим образом 

ID      Key  Value  Object
1001    K    1        4
1001    C    2        7
1001    D    3        9
2001    K    1        19
2001    C    2        10

Теперь я хотел бы свернуть фрейм данных следующим образом: 

ID   Value_K   Value_C   Value_D  Object_K  Object_C Object_D
1001   1         2         3         4         7        9
2001   1         2         NA        19        10       NA

Может ли кто-то быть так любезен, чтобы помочь мне.Я пытался с pivot (), pivot_table (), collapse () и melt (), но не достиг никаких результатов.

Спасибо

1 Ответ

1 голос
/ 16 марта 2019

Вы можете использовать groupby и unstack

df = pd.DataFrame({'ID': [1001,1001,1001,2001,2001], 'Key': ['K', 'C', 'D', 'K', 'C'], 'Value': [1,2,3,1,2],
                  'Object': [4,7,9,19,10]})
df = df.groupby(['ID', 'Key']).sum().unstack()
df.columns = df.columns.map('{0[0]}-{0[1]}'.format)
df = df.reset_index()
print(df)

     ID  Value-C  Value-D  Value-K  Object-C  Object-D  Object-K
0  1001      2.0      3.0      1.0       7.0       9.0       4.0
1  2001      2.0      NaN      1.0      10.0       NaN      19.0
...