pandas 집계만들기

import pandas

data = pandas.read_csv("C:\\Users\\PC\\Desktop\\erx\\ex_4\\2018_Central_Park_Squirrel_Census_-_Squirrel_Data_20240617.csv")

grey_squirrels_count =len(data[data["Primary Fur Color"] =="Gray"])

red_squirrels_count =len(data[data["Primary Fur Color"] =="Cinnamon"])

black_squirrels_count =len(data[data["Primary Fur Color"] =="Black"])

print(grey_squirrels_count)

print(red_squirrels_count)

print(black_squirrels_count)

data_dict ={

    "fur Color":["Grey","Cinnamon","Black"],

    "Count":[grey_squirrels_count,red_squirrels_count,black_squirrels_count]

}

df=pandas.DataFrame(data_dict)

df.to_csv("color_count.csv")