Создать новые столбцы в кадре данных с помощью комбинации трех столбцов в Python

Question

У меня есть что-то вроде этого:

       Date  Id  Product  Sales
0  1/1/2001   1       21      1200
1  1/1/2001   1       22      1000
2  1/1/2001   1       23      1500
3  2/1/2001   1       21      300
4  2/1/2001   2       22      200
5  3/1/2001   3       21      400
6  4/1/2001   3       22      500

Я хочу создать что-то подобное с той же таблицей:

Answer 1

Вы должны объединить ID и Product, а затем повернуть результат.

import pandas as pd

df=pd.DataFrame([['1/1/2001',1,21,1200],['1/1/2001',1,22,1000],['1/1/2001',1,23,1500],['2/1/2001',1,21,300],['2/1/2001',2,22,200],['3/1/2001',3,21,400],['4/1/2001',3,22,500]],columns=('Date','Id','Product','Sales'))

df['Id_Prod'] = df['Id'].astype(str).str.cat(df['Product'].astype(str), sep='_')

df.pivot(index='Date',columns='Id_Prod',values='Sales')

Результат:

Answer 2

Это легко сделать с помощью функции Pivot Pandas.

Итак, это ваш Dataframe:

df=pd.DataFrame([['1/1/2001',1,21,1200],['1/1/2001',1,22,1000],['1/1/2001',1,23,1500],['2/1/2001',1,21,300],['2/1/2001',2,22,200],['3/1/2001',3,21,400],['4/1/2001',3,22,500]],columns=('Date','Id','Product','Sales'))

Вывод:

    Date    Id  Product Sales
0   1/1/2001    1   21  1200
1   1/1/2001    1   22  1000
2   1/1/2001    1   23  1500
3   2/1/2001    1   21  300
4   2/1/2001    2   22  200
5   3/1/2001    3   21  400
6   4/1/2001    3   22  500

Теперь просто используйтеэтот код:

df.pivot(index='Date',columns='Product',values='Sales')

И вы получите:

Product      21     22      23
Date            
1/1/2001    1200.0  1000.0  1500.0
2/1/2001    300.0   200.0   NaN
3/1/2001    400.0   NaN     NaN
4/1/2001    NaN     500.0   NaN

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