gpt4 book ai didi

python - 如何以不同的速率从 pandas 数据框中对每个组进行采样

转载 作者:太空宇宙 更新时间:2023-11-03 13:55:59 25 4
gpt4 key购买 nike

我有一个数据框,其中包含我希望从中生成样本的人口信息。我还有一个数据框 sample_info,它详细说明了我的示例中需要的 population 数据框中每个组的单位数。我已经开发了一些代码来实现我所需要的,但考虑到我正在使用的大型数据集,它的运行速度比我希望的要慢。

有没有一种方法可以对人口框架进行分组并将抽样应用于各组,而不是像我在下面所做的那样循环遍历它们?

import pandas as pd

population = pd.DataFrame([[1,True],[1,False],[1,False],[2,True],[2,True],[2,False],[2, True]], columns = ['Group ID','Response'])

Group ID Response
0 1 True
1 1 False
2 1 False
3 2 True
4 2 True
5 2 False
6 2 True

sample_info = pd.DataFrame([[1,5],[2,6]], columns = ['Group ID','Sample Size'])

output = pd.DataFrame(columns = ['Group ID','Response'])

Group ID Sample Size
0 1 5
1 2 6


for index, row in sample_info.iterrows():
output = output.append(population.loc[population['Group ID'] == row['Group ID']].sample(n=row['Sample Size'], replace = True))

我想不出如何使用 group-by 引入样本大小信息并按照 Pandas: sample each group after groupby 中的建议应用

最佳答案

将 sample_info 转换为字典。按组 ID 分组。使用字典将样本大小值传递给 DataFrame.sample。

mapper = sample_info.set_index('Group ID')['Sample Size'].to_dict()

population.groupby('Group ID').apply(lambda x: x.sample(n=mapper.get(x.name))).reset_index(drop = True)

关于python - 如何以不同的速率从 pandas 数据框中对每个组进行采样,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55523917/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com