Выполните Pandas Goupby и процентное увеличение / уменьшение - PullRequest
Новая
       2

Выполните Pandas Goupby и процентное увеличение / уменьшение

2 голосов
/ 13 апреля 2020

У меня есть датафрейм, как показано ниже: 

           Date      Symbol      Quantity        Volume
0    07-04-2020         ACC       1579268        554047
1    08-04-2020         ACC       2153627        586886
2    09-04-2020         ACC       2658885       1128891
3    07-04-2020    ADANIENT       2463525        408903
4    08-04-2020    ADANIENT       3237290        255480
5    09-04-2020    ADANIENT       2359528        335061

Я хочу выполнить групповую обработку по символу и рассчитать процентное увеличение или уменьшение с 1-й даты, например (07-04-2020 в моем примере) для последующих строк .

Я хочу результат, как показано ниже: 

    Date         Symbol    Quantity  Volume  PctCH_Qty    PctCH_Vol
0   07-04-2020      ACC     1579268  554047          0           0   
1   08-04-2020      ACC     2153627  586886      36.37        5.93 
2   09-04-2020      ACC     2658885  1128891     68.36      103.75 
3   07-04-2020  ADANIENT    2463525  408903          0           0   
4   08-04-2020  ADANIENT    3237290  255480      31.41      -37.52 
5   09-04-2020  ADANIENT    2359528  335061      -4.22      -18.06

1 Ответ

2 голосов
/ 13 апреля 2020

Идея состоит в том, чтобы разделить первое значение по группам на GroupBy.transform и first, вычесть 1, кратное 100 и последний раунд: 

cols = ['Quantity','Volume']
df[['PctCH_Qty','PctCH_Vol']] = (df[cols].div(df.groupby('Symbol')[cols]
                                         .transform('first'), axis=0)
                                         .sub(1)
                                         .mul(100)
                                         .round(2))
print (df)
         Date    Symbol  Quantity   Volume  PctCH_Qty  PctCH_Vol
0  07-04-2020       ACC   1579268   554047       0.00       0.00
1  08-04-2020       ACC   2153627   586886      36.37       5.93
2  09-04-2020       ACC   2658885  1128891      68.36     103.75
3  07-04-2020  ADANIENT   2463525   408903       0.00       0.00
4  08-04-2020  ADANIENT   3237290   255480      31.41     -37.52
5  09-04-2020  ADANIENT   2359528   335061      -4.22     -18.06
...