基于唯一值的随机行

Question

我想从 DF 中获取 2 个随机但不同的人员 ID，并将其放入另一个 DF

Person_id	邮政	积极的
567	是的	不活跃
678	是的	积极的
567	不	不活跃
689	是的	积极的
680	是的	不活跃
689	不	积极的

df['person_id'].sample(n=100, random_state=1)

此代码未获取唯一的 person_id 值，仅将该列放在 df 中。 我需要获取该特定列的许多唯一值，并将其与所有其他列一起放入 df 中。

df.person_id.sample(n=100, random_state=1).groupby('person_id')

我也试过这个，但它创建了一个奇怪的 object

有小费吗？

Answer 1

df = pd.DataFrame({'person_id': [567, 678, 567, 689, 680, 689],
                   'post': ['yes', 'yes', 'no', 'yes', 'yes', 'no'],
                   'active': ['inactive', 'active', 'inactive', 'active', 'inactive', 'active']})

到 select 两个随机的唯一人员 ID：

selected = df['person_id'].drop_duplicates().sample(n=2)

要为选定的人员 ID 创建包含所有行的数据框：

df[df['person_id'].isin(selected)]