Lab/CDC_vital_statistics.py at master · NYUDataBootcamp/Lab · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
"""
CDC vital statistics

DOES NOT WORK YET

Link
* http://www.cdc.gov/nchs/data_access/vitalstatsonline.htm

Prepared for Data Bootcamp course at NYU
* https://github.com/NYUDataBootcamp/Materials
* https://github.com/NYUDataBootcamp/Materials/Code/Lab

Written by Dave Backus, January 2016
Created with Python 3.5
"""
"""
import packages, check versions
"""
import sys
import pandas as pd
#import matplotlib.pyplot as plt

print('\nPython version: ', sys.version)
print('Pandas version: ', pd.__version__, '\n')

#%%
"""
read data
"""
url1 = 'ftp://ftp.cdc.gov/pub/Health_Statistics/NCHS/Datasets/'
url2 = 'DVS/mortality/mort2014us.zip'

# healthcare spending
cdc = pd.read_excel(url1+url2,
                   skiprows=3, sheetname=1, index_col=0,
                   na_values=['..'])


#%%
# OLD STUFF, IGNORE
# select years
hc = hc[list(range(1980,2014))]
# select countries and transpose df
countries = ['Canada', 'France', 'Germany', 'Japan', 'United Kingdom',
             'United States']
some = hc[hc.index.isin(countries)].T

# plot
ax = some.plot(lw=2, subplots=False)
ax.set_title('Healthcare spending', fontsize=14, loc='left')
ax.set_ylabel('Percent of GDP')
ax.legend(loc='upper left', fontsize=10, handlelength=2, labelspacing=0.15)

#%%
"""
Plot number of docs
"""
# number of docs
docs = pd.read_excel(url1+url2,
                   skiprows=3, sheetname='Physicians', index_col=0,
                   skip_footer=21,
                   na_values=['..'])

# select years
docs = docs[[2012]]
# mpd.columns = map(str.rstrip, mpd.columns)
docs.index = [name.rsplit(maxsplit=1)[0] for name in docs.index.tolist()]

# select countries
countries = ['Canada', 'France', 'Germany', 'Japan', 'United Kingdom',
             'United States']
some = docs[docs.index.isin(countries)]

# plot
ax = some.plot(kind='barh', alpha=0.5, legend=False)
ax.set_title('Number of doctors', fontsize=14, loc='left')
ax.set_xlabel('Number of Doctors per 1000 Population')