У меня две мультииндексные серии

Серия 1

Company Name     Product     Price        TransactionID
Company A        Apple       10           T0001
Company B        Grapes      20           T0002
                 Orange      30           T0003

Серия 2

Company Name     Product     Price        TransactionID
Company A        Orange      10           T0004
                 Apple       20           T0005
Company B        Orange      20           T0006

Я хочу сформировать серию, как показано ниже

Company Name     Product     Sum_price    TransactionID
Company A        Orange      10           [T0004]
                 Apple       30           [T0001,T0005]
Company B        Orange      50           [T0003,T0006]
                 Grapes      20           [T0002]

Из фрейма данных группы и получения суммы И подсчета? показано, что я могу использовать .agg для суммирования и подсчета одновременно, но как применить .apply (list) и суммировать вместе? Прямо сейчас я применил код Pandas: добавление мультииндексных серий / Dataframes, содержащих списки для формирования двух серий: одна - сумма, а другая - с идентификатором транзакции. Я думаю, что могу объединить две серии вместе, но я просто хочу посмотреть, есть ли лучший способ сделать это.

2
Platalea Minor 23 Окт 2018 в 13:04

2 ответа

Лучший ответ

Используйте concat с агрегатом по < a href = "http://pandas.pydata.org/pandas-docs/stable/generated/pandas.core.groupby.GroupBy.aggregate.html" rel = "nofollow noreferrer"> aggregate с sum и преобразованием в list:

df = (pd.concat([df1, df2])
        .groupby(['Company Name', 'Product'], as_index=False)
        .agg({'Price':'sum', 'TransactionID': lambda x: x.tolist()})
     )
print (df)
  Company Name Product  Price   TransactionID
0    Company A   Apple     30  [T0001, T0005]
1    Company A  Orange     10         [T0004]
2    Company B  Grapes     20         [T0002]
3    Company B  Orange     50  [T0003, T0006]
1
jezrael 23 Окт 2018 в 10:41

Сначала вы можете объединить оба ваших фрейма данных следующим образом:

df3 = pd.concat([df1, df2])

После этого вы можете использовать groupby agg для суммирования, а также использовать лямбда-функцию для объединения списков.

df3 = df3.groupby(['Company', 'Name', 'Product']).agg({'Price':sum, 'TransactionID': lambda x: [i for i in x]})

После этого вы получите желаемый результат:

Company Name     Product     Sum_price    TransactionID
Company A        Orange      10           [T0004]
                 Apple       30           [T0001,T0005]
Company B        Orange      50           [T0003,T0006]
                 Grapes      20           [T0002]
1
zrelova 23 Окт 2018 в 10:25
52946348