[PYTHON] Plotting with matplotlib

plotting.py

import matplotlib.pyplot as plt
import seaborn as sns

gdp_cap = [10, 11, 12, 14, 18]
life_exp = [11, 30, 13, 41, 12]
# Specify c and alpha inside plt.scatter()
plt.scatter(x = gdp_cap, y = life_exp, s = np.array(pop) * 2, c = col, alpha = 0.8)

# Previous customizations
plt.xscale('log') 
plt.xlabel('GDP per Capita [in USD]')
plt.ylabel('Life Expectancy [in years]')
plt.title('World Development in 2007')
plt.xticks([1000,10000,100000], ['1k','10k','100k'])
plt.xticks(rotation=90)

# Additional cool customizations
plt.figure()
sns.countplot(x='education', hue='party', data=df, palette='RdBu')
plt.xticks([0,1], ['No', 'Yes'])
plt.show()

# Show the plot
plt.show()