warunkowe kwoty dla agregatów pand
Niedawno dokonałem przełączenia z R na python i miałem problemy z przyzwyczajeniem się do ramek danych w przeciwieństwie do używania danych R.table. Problem, z którym miałem do czynienia, polega na tym, że chciałbym pobrać listę ciągów, sprawdzić wartość, a następnie zliczyć liczbę tego ciągu - podzieloną przez użytkownika. Chciałbym więc wziąć te dane:
A_id B C
1: a1 "up" 100
2: a2 "down" 102
3: a3 "up" 100
3: a3 "up" 250
4: a4 "left" 100
5: a5 "right" 102
I wróć:
A_id_grouped sum_up sum_down ... over_200_up
1: a1 1 0 ... 0
2: a2 0 1 0
3: a3 2 0 ... 1
4: a4 0 0 0
5: a5 0 0 ... 0
Zanim to zrobiłem z kodem R (używając data.table)
>DT[ ,list(A_id_grouped, sum_up = sum(B == "up"),
+ sum_down = sum(B == "down"),
+ ...,
+ over_200_up = sum(up == "up" & < 200), by=list(A)];
Jednak wszystkie moje ostatnie próby z Pythonem zawiodły mnie:
DT.agg({"D": [np.sum(DT[DT["B"]=="up"]),np.sum(DT[DT["B"]=="up"])], ...
"C": np.sum(DT[(DT["B"]=="up") & (DT["C"]>200)])
})
Z góry dziękuję! wydaje się to proste pytanie, ale nie mogłem go nigdzie znaleźć.