如何使用 pandas 数据框构建人口金字塔

Question

如何根据以下起始数据框绘制人口金字塔？

           Age  Gender  Count
0  50-45 years    male      4
1  50-45 years  female      5
2  55-65 years    male      6
3  55-65 years  female      7
4  65-70 years    male     11
5  65-70 years  female     12

我尝试了以下方法，，但结果图看起来很奇怪：

import pnadas as pd
import seaborn as sns

# data
data = {'Age': ['50-45 years', '50-45 years', '55-65 years', '55-65 years', '65-70 years', '65-70 years'],
        'Gender': ['male', 'female', 'male', 'female', 'male', 'female'], 'Count': [4, 5, 6, 7, 11, 12]}

df = pd.DataFrame(data)

# plot
sns.barplot(data=df, x='Count', y='Age',
            hue='Gender', orient='horizontal', 
            dodge=False)

我认为问题是我的年龄是一个字符串。

Answer 1

与链接的问题不同，'Gender' 组的 'Count' 都是正数，因此对于 dodge=False，'Female' 条形图绘制在 'Male' 条。
使用 .loc 和布尔选择将其中一组转换为负值。

# convert male counts to negative
df.loc[df.Gender.eq('male'), 'Count'] = df.Count.mul(-1)

# plot
sns.barplot(data=df, x='Count', y='Age', hue='Gender', orient='horizontal', dodge=False)

如何使用 pandas 数据框构建人口金字塔

How to build a population pyramid with pandas dataframe

python

matplotlib

bar-chart

pandas

seaborn