import pandas_datareader.wb as worldbank
import matplotlib.pyplot as plt
import pandas as pd
plt.rcParams['font.sans-serif']=['SimHei']
start_year=2003
end_year=2023
dat=worldbank.download(indicator='NY.GDP.PCAP.KD',country=['USA','SWE','CHE'],\
start=start_year,end=end_year)
print(dat)
dat2draw=dat.unstack(level=0)
plt.plot(dat2draw.iloc[:,0],'r-',label='USA')
plt.plot(dat2draw.iloc[:,1],'b-*',label='SWE')
plt.plot(dat2draw.iloc[:,2],'g--',label='CHE')
plt.title('人均GDP对比',fontsize=20)
plt.legend()
plt.xticks(fontsize=8,rotation=60)
plt.show()

