CODEWITHSUNDEEP
pythonpandassumrow
UserYmY
UserYmY

Reputation: 8574

Python Pandas: Counting the frequency of a specific value in each row of dataframe?

I have a dataframe df:

domain               country     out1 out2 out3
oranjeslag.nl           NL          1    0   NaN    
pietervaartjes.nl       NL          1    1    0
andreaputting.com.au    AU          NaN  1    0 
michaelcardillo.com     US          0    0    NaN

I would like to define two columns sum_0 and sum_1 and count the number of 0s and 1s in columns (out1,out2,out3),per row. So expected results would be:

domain               country     out1 out2 out3   sum_0  sum_1
oranjeslag.nl           NL          1    0   NaN    1      1
pietervaartjes.nl       NL          1    1    0     1      2
andreaputting.com.au    AU          NaN  1    0     1      1
michaelcardillo.com     US          0    0    NaN   2      0

I have this code for counting the number of 1s, but I do not know how to count the number of 0s.

df['sum_1'] = df[['out_1','out_2','out_3']].sum(axis=1)

Can anybody help?

Upvotes: 6

Views: 24610

Answers (3)

Tomas Trdla
Tomas Trdla

Reputation: 1202

maybe pandas changed the bahaviour since 2015 but now the problem with sum is, that when you try to use this code for values > 1, it produces actual sum of these values, not their count (which is what I understood from question and also was looking for)

df['sum_0'] = df[df == 0].count(axis=1)

Upvotes: 1

EdChum
EdChum

Reputation: 394419

You can call sum for each condition, the 1 condition is simple just a straight sum on axis=1, for the second you can compare the df against 0 value and then call sum as before:

In [102]:
df['sum_1'] = df[['out1','out2','out3']].sum(axis=1)
df['sum_0'] = (df[['out1','out2','out3']] == 0).sum(axis=1)
df

Out[102]:
                 domain country  out1  out2  out3  sum_0  sum_1
0         oranjeslag.nl      NL     1     0   NaN      1      1
1     pietervaartjes.nl      NL     1     1     0      1      2
2  andreaputting.com.au      AU   NaN     1     0      1      1
3   michaelcardillo.com      US     0     0   NaN      2      0

Upvotes: 10

AntonyBrd
AntonyBrd

Reputation: 403

I would do :

df["sum_0"] = df.apply(lambda row: sum(row[0:3]==0) ,axis=1)

Upvotes: 5

Related Questions