2013-08-29 18 views
21

Oto uproszczony przykład mojego df:pandy dataframe tworzyć nowe kolumny i napełnić obliczonych wartości od samego df

ds = pd.DataFrame(np.abs(randn(3, 4)), index=[1,2,3], columns=['A','B','C','D']) 
ds 
     A   B   C   D 
1 1.099679 0.042043 0.083903 0.410128 
2 0.268205 0.718933 1.459374 0.758887 
3 0.680566 0.538655 0.038236 1.169403 

chciałbym podsumować dane w kolumnach row mądry:

ds['sum']=ds.sum(axis=1) 
ds 
     A   B   C   D  sum 
1 0.095389 0.556978 1.646888 1.959295 4.258550 
2 1.076190 2.668270 0.825116 1.477040 6.046616 
3 0.245034 1.066285 0.967124 0.791606 3.070049 

Oto moje pytanie! Chciałbym utworzyć 4 nowe kolumny i obliczyć procentową wartość z sumy (sumy) w każdym wierszu. Tak więc pierwsza wartość w pierwszej nowej kolumnie powinna wynosić (0,095389/4,258550), pierwsza wartość w drugiej nowej kolumnie (0,556978/4,258550) ... i tak dalej ... Pomoc proszę

Odpowiedz

37

Możesz to łatwo zrobić ręcznie dla każdej kolumny jak ten:

df['A_perc'] = df['A']/df['sum'] 

Jeśli chcesz to zrobić w jednym kroku dla wszystkich kolumn, można użyć metody div (http://pandas.pydata.org/pandas-docs/stable/basics.html#matching-broadcasting-behavior):

ds.div(ds['sum'], axis=0) 

A jeśli chcesz to w jednym kroku dodanego do tej samej dataframe:

>>> ds.join(ds.div(ds['sum'], axis=0), rsuffix='_perc') 
      A   B   C   D  sum A_perc B_perc \ 
1 0.151722 0.935917 1.033526 0.941962 3.063127 0.049532 0.305543 
2 0.033761 1.087302 1.110695 1.401260 3.633017 0.009293 0.299283 
3 0.761368 0.484268 0.026837 1.276130 2.548603 0.298739 0.190013 

    C_perc D_perc sum_perc 
1 0.337409 0.307517   1 
2 0.305722 0.385701   1 
3 0.010530 0.500718   1 
+0

Dziękuję !! Właśnie tego chciałem – jonas

+1

Robiąc coś takiego, czy pandy gwarantują, że kolejność rzędów się nie zmieni? – kalu

+0

Tak, kolejność rzędów jest z pewnością zachowana. – joris

2
In [56]: df = pd.DataFrame(np.abs(randn(3, 4)), index=[1,2,3], columns=['A','B','C','D']) 

In [57]: df.divide(df.sum(axis=1), axis=0) 
Out[57]: 
      A   B   C   D 
1 0.319124 0.296653 0.138206 0.246017 
2 0.376994 0.326481 0.230464 0.066062 
3 0.036134 0.192954 0.430341 0.340571 
+1

Dla przejrzystości, 'div' i' divide' są takie same (tylko aliasy) – joris