mirror of
https://github.com/wahyd4/code-sandbox.git
synced 2026-08-08 21:03:28 +10:00
25 lines
682 B
Python
25 lines
682 B
Python
import pandas as pd
|
|
import numpy as np
|
|
|
|
populations = pd.read_csv("data/state-population.csv")
|
|
areas = pd.read_csv("data/state-areas.csv")
|
|
abbrevs = pd.read_csv("data/state-abbrevs.csv")
|
|
|
|
merged = pd.merge(
|
|
populations, abbrevs, how="outer", left_on="state/region", right_on="abbreviation"
|
|
)
|
|
|
|
merged = merged.drop("abbreviation", 1)
|
|
|
|
final = pd.merge(merged, areas, on="state", how="left")
|
|
|
|
final.dropna(inplace=True)
|
|
|
|
data2010 = final.query("year == 2010 & ages == 'total'")
|
|
# print(data2010.head())
|
|
|
|
data2010.set_index("state", inplace=True)
|
|
density = data2010["population"] / data2010["area (sq. mi)"]
|
|
density.sort_values(ascending=False, inplace=True)
|
|
print(density.tail())
|