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']])