import pandas
data = pandas.read_csv("2018_Central_Park_Squirrel_Census_-_Squirrel_Data.csv")
fur_color_data = data["Primary Fur Color"]
# Alternative solution: value_counts()
count = fur_color_data.value_counts()
print(count)
squirrels_count = pandas.DataFrame(count)
squirrels_count.to_csv("squirrels_count.csv")
# Teacher's solution
grey_squirrels_count = len(data[data["Primary Fur Color"] == "Grey"])
red_squirrels_count = len(data[data["Primary Fur Color"] == "Cinnamon"])
black_squirrels_count = len(data[data["Primary Fur Color"] == "Black"])
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("squirrels_count1.csv")