다람쥐 분류하기

목표

https://data.cityofnewyork.us/Environment/2018-Central-Park-Squirrel-Census-Squirrel-Data/vfnx-vebw

위 사이트에서 센트럴 파크의 다람쥐 데이터를 다운로드하고 털 색에 따라서 다람쥐를 분류해보자

코드

import pandas

data = pandas.read_csv("2018_Central_Park_Squirrel_Census_-_Squirrel_Data.csv")

color = data["Primary Fur Color"]

result = pandas.DataFrame({
    "Fur Color":["grey", "red", "black"],
    "Count":[len(data[color=="Gray"]), len(data[color=="Cinnamon"]), len(data[color=="Black"])]
})

result.to_csv("result.csv")

실행 결과