python: How to count the elements of a row?

Question

For example, I have a DataFrame named a. I want to count the element of each row.

import numpy as np a=pd.DataFrame({'A1':['financial','game','game'],'A2':['social','food','sport'],'A3':['social','sport','game']})

Input:

 A1 A2 A3 0 financial social social 1 game food sport 2 game sport game

Expected:

 financial food game social sport 0 1 0 0 2 0 1 0 1 1 0 1 2 0 0 2 0 1

Hopefully for help, thanks!

jezrael · Accepted Answer · 2018-09-14 12:01:26Z

df = pd.get_dummies(a, prefix_sep='', prefix='').sum(axis=1, level=0) print (df) financial game food social sport 0 1 0 0 2 0 1 0 1 1 0 1 2 0 2 0 0 1

df = a.stack().groupby(level=0).value_counts().unstack(fill_value=0) print (df) financial food game social sport 0 1 0 0 2 0 1 0 1 1 0 1 2 0 0 2 0 1

1 Answer 1