人口金字塔,包括巨蟒和海上金字塔
问题描述
我正在尝试创建一个按性别分组的人口金字塔。不幸的是,我不能让它工作。情节只是一幅白色的图画,轴线似乎在某种程度上颠倒了过来。也许有人能帮我,谢谢。
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
# I read this testdata from a csv file
testdata = pd.DataFrame({'age': [20, 20, 21, 21, 22, 22, 23, 23],
'gender': ["male", "female", "male", "female", "male", "female", "male", "female"],
'count': [10, -12, 13, -10, 16, -14, 17, -16]});
plt.figure(figsize=(13, 10), dpi=80)
group_col = 'gender'
order_of_bars = testdata['age'].unique()[::-1]
colors = [plt.cm.Spectral(i / float(len(testdata[group_col].unique()) - 1)) for i in range(len(testdata[group_col].unique()))]
for c, group in zip(colors, testdata[group_col].unique()):
barplot = sns.barplot(x='count', y='age', data=testdata.loc[testdata[group_col] == group, :], order=order_of_bars, color=c, label=group)
plt.xlabel("Counts")
plt.ylabel("Age")
plt.yticks(fontsize=12)
plt.title("Pyramide", fontsize=22)
plt.legend()
plt.show()
解决方案
如果您要查找this population pyramid,我们试试:
sns.barplot(data=testdata, x='count',y='age',
hue='gender',orient='horizontal',
dodge=False)
输出:
相关文章