python-区间划分,统计年龄段性别占比

问题描述:

有个DataFrame(gender,age),需要按指定间隔划分不同分组,并求该分组的性别占比(男性+女性=100%)
间隔:bin = [0, 20, 40, 60, 100],左开右闭

知识点

1、通过该例可以学习random的使用
2、学习pd.cut划分数值区间
3、pd.agg()

解答:

# 1、构造测试数据
random.seed(1)  # 指定随机种子,保证每次运行结果一致
df = pd.DataFrame(
    {'gender': [random.choice([1, 0]) for _ in range(100)],
     'age': [random.randint(1, 100) for _ in range(100)]}
)
# 2、将年龄划分不同时间段
age_bin = [0, 20, 40, 60, 100]
df['layered'] = pd.cut(x=df['age'], bins=age_bin)
# 3、将age划分对应区间
df = df.groupby(['layered']).agg({'gender': 'sum', 'age': 'count'})

# 4、计算男女占比情况
df = df.assign(
    male_rate=df['gender'] / df['age'],
    female_rate=1 - df['gender'] / df['age']
)
# 5、输出结果
print(df[['male_rate', 'female_rate']])

python-区间划分,统计年龄段性别占比_第1张图片

你可能感兴趣的:(数值划分区间,python,python)