python - 如何在多个列上进行 Pandas 样本？

Question

我有一个包含大约 800 万个观察值的数据框。我需要从中提取样本，但想从多个列中采样。

我尝试了以下方法，但不起作用:

import pandas as pd

state = ['mi', 'mi', 'mi', 'nc', 'pa', 'pa', 'ga']
state = state * 50
age = ['21', '22', '23', '23', '23', '50', '50']
age = age * 50
random = ['.445', '.324', '.234', '.143', '.568', '.777', '.256']
random = random * 50
data = {'state':state, 'age': age, 'random': random}
df = pd.DataFrame.from_dict(data = data)

df_sample = df.sample(n = 25, weights = ['state', 'age'], random_state = 48)

我意识到pandas文档没有说明我想做的事情是可能的。有什么办法可以做到这一点吗？

Answer 1

IIUC，

我认为您希望实现以下目标:

df_sample = df[['state','age']].sample(n = 25, random_state = 48)