pps抽样python代码

PPS（每千人样本）抽样是一种统计抽样方法，通常用于人口普查或市场研究中，目的是确保每个样本代表性的概率与其在总体中的相对大小成比例。在Python中，我们可以使用numpy库来进行这样的抽样。这里是一个简单的例子：

import numpy as np

defpps_sampling(total_population, sample_size):
    # 计算每个个体的权重，即1000除以总人口，然后乘以样本大小
    weights = (sample_size / total_population) * 1000
    
    # 使用numpy的random.choice函数，设置replace=False（不允许重复抽样），weight参数为计算出的权重
    pps_sample = np.random.choice(total_population, size=sample_size, replace=False, p=weights)
    
    return pps_sample

# 使用示例
total_people = 1000000
desired_sample = 500
pps_sampled_people = pps_sampling(total_people, desired_sample)

print(f"PPS抽样的样本：{pps_sampled_people}")