Как получить строку из имен столбцов и объединить несколько столбцов в python?

#python #melt

Вопрос:

У меня есть эта таблица, и мне нужно расплавить эту таблицу, чтобы она была похожа на ожидаемую таблицу, в которой мне нужно получить имя точки (a и b) из имени столбца и позволить столбцам bq и прогресса расплавиться.

 type    bq a    bq b    progress a    progress b
P        1       1          1             2
Q        2       3          4             2
R        2       1          1             2
 

Ожидаемый результат выглядит следующим образом:

 type     point      bq    progress
P         a         1        1
P         b         1        2
Q         a         2        4
Q         b         3        2
R         a         2        1
R         b         1        2
 

Как это можно сделать на python?

Ответ №1:

Попробуйте это:

 df = pd.DataFrame({'type':['p','q','r'],
                  'bq a':['1','2','2'],
                  'bq b':['1','3','1'],
                  'progress a':['1','4','1'],
                  'progress b':['2','2','2']})
 

введите описание изображения здесь

 df_bq = pd.melt(df, id_vars =['type'], value_vars =['bq a','bq b'])
df_bq.columns = ['type','point','bq']
df_bq['point'] = df_bq['point'].apply(lambda x:x.replace('bq ',''))
df_bq.sort_values(by = 'point')

df_p = pd.melt(df, id_vars =['type'], value_vars =['progress a','progress b'])
df_p.columns = ['type','point','progress']
df_p['point'] = df_p['point'].apply(lambda x:x.replace('progress ',''))
df_p.sort_values(by = 'point')

df_concat = pd.concat([df_bq, df_p['progress']], axis=1)
df_concat
 

Результат :

введите описание изображения здесь