Преобразование последовательных столбцов в соответствующие строки в Pandas DataFrame

avatar
Narendra Prasath
1 июля 2021 в 17:49
50
1
2

Попытка преобразовать последовательные столбцы в строки в pandas. Пример: последовательные имена столбцов представляют собой последовательные числа вместе с некоторыми строками, т.е. > в DataFrame. Вы можете использовать приведенный ниже код для создания кадра данных.

df = pd.DataFrame({'City': ['Houston', 'Austin', 'Hoover'],'State': ['Texas', 'Texas', 'Alabama'],'Name':['Aria', 'Penelope', 'Niko'],'Key1':["test1", "test2", "test3"],'Val1':[28, 4, 7],'Key2':["test4", "test5", "test6"],
'Val2':[82, 45, 76],'Key3':["test7", "test8", "test9"],'Val3':[4, 76, 9],'Key4':["test10", "test11", "test12"],'Val4':[97, 66, 10],'Key5':["test13", "test14", "test15"],'Val5':[4, 10, '']},columns=['City', 'State', 'Name', 'Key1', 'Val1', 'Key2', 'Val2', 'Key3', 'Val3', 'Key4', 'Val4', 'Key5', 'Val5'])

enter image description here

Я попробовал функцию melt, как показано ниже:

df.melt(id_vars=['City', 'State'], var_name='Column', value_name='Key')

Но я получил следующий вывод:

enter image description here

Проблема в том, что для каждого ключа столбец val имеет разные строки. ожидаемый результат приведен ниже:

enter image description here

Источник

Ответы (1)

avatar
Scott Boston
1 июля 2021 в 17:57
5

Используйте pd.wide_to_long:

pd.wide_to_long(df,['Key', 'Val'],['City', 'State', 'Name'],'No').reset_index()

Вывод:

       City    State      Name  No     Key Val
0   Houston    Texas      Aria   1   test1  28
1   Houston    Texas      Aria   2   test4  82
2   Houston    Texas      Aria   3   test7   4
3   Houston    Texas      Aria   4  test10  97
4   Houston    Texas      Aria   5  test13   4
5    Austin    Texas  Penelope   1   test2   4
6    Austin    Texas  Penelope   2   test5  45
7    Austin    Texas  Penelope   3   test8  76
8    Austin    Texas  Penelope   4  test11  66
9    Austin    Texas  Penelope   5  test14  10
10   Hoover  Alabama      Niko   1   test3   7
11   Hoover  Alabama      Niko   2   test6  76
12   Hoover  Alabama      Niko   3   test9   9
13   Hoover  Alabama      Niko   4  test12  10
14   Hoover  Alabama      Niko   5  test15    

Вы пытаетесь одновременно расплавить две колонки. pd.wide_to_long обрабатывает эту ситуацию.