move python code snippets to python folder

This commit is contained in:
2019-06-22 22:50:39 +10:00
parent b76f2e25bd
commit 2c8848211b
13 changed files with 0 additions and 0 deletions
+24
View File
@@ -0,0 +1,24 @@
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())