利用分位数生成子集

import pandas as pd

# 示例数据集
data = {
    '姓名': ['张三', '李四', '王五', '赵六', '孙七', '周八', '吴九', '郑十'],
    '收入': [5000, 8000, 12000, 15000, 20000, 25000, 30000, 35000]
}
df = pd.DataFrame(data)

# 计算收入列的四分位数
q1 = df['收入'].quantile(0.25)
q2 = df['收入'].quantile(0.5)
q3 = df['收入'].quantile(0.75)

# 定义子集标准并生成子集
subset1 = df[df['收入'] <= q1]  # 最低收入群体
subset2 = df[(df['收入'] > q1) & (df['收入'] <= q2)]  # 中等偏低收入群体
subset3 = df[(df['收入'] > q2) & (df['收入'] <= q3)]  **中等偏高收入群体**
subset4 = df[df['收入'] > q3]  # 最高收入群体

# 输出子集
print("最低收入群体：")
print(subset1)
print("\n中等偏低收入群体：")
print(subset2)
print("\n中等偏高收入群体：")
print(subset3)
print("\n最高收入群体：")
print(subset4)