StackBartender
StackBartender

Reputation: 31

python count rows from csv pandas

colors.csv

id  name    rgb         is_trans
0   -1  Unknown 0033B2  f
1   0   Black   05131D  f
2   1   Blue    0055BF  t

How do you count how many f & t, (something like below)

colors_summary = colors.count('is_trans')
print(colors_summary)

Looking for Result

is_trans    id  name    rgb
f   107 107 107
t   28  28  28

Upvotes: 1

Views: 884

Answers (2)

ahed87
ahed87

Reputation: 1360

and an alternative with stdlib csv & Counter.

color_csv = """id  name    rgb         is_trans
0   -1  Unknown 0033B2  f
1   0   Black   05131D  f
2   1   Blue    0055BF  t"""

import csv
from collections import Counter
from io import StringIO

settings = dict(delimiter=' ', skipinitialspace=True)
creader = csv.reader(StringIO(color_csv), **settings)
headers = next(creader)
counter = Counter((row[-1] for row in creader))
print(counter)

Counter({'f': 2, 't': 1})

Upvotes: 0

usher
usher

Reputation: 86

let say you have

color_df # dataframe object

you can do that:

result_df = color_df.groupby('is_trans').count()
print(result_df) # should give you what you ask for.

Upvotes: 1

Related Questions