Python：随机选择组中的子组

Question

我有一个看起来像这样的数据框：

patient_id   note_id    lines
A              10         1
A              10         2
A              10         3
A              29         1
A              29         2 
B              12         1
B              95         1
B              95         2
B              95         3
C......
D...... 
E              14         1
E              55         1 
E              87         1
......

每个患者可以有多个注释，每个注释可能包含多于 1 行。假设我有 20 个患者，50 个注释和 150 行。如何为随机选择的 3 位患者随机选择一个随机注释？假设我希望每个随机选择的患者 ID 有一个随机注释，我会得到：

patient_id   note_id    lines
A              29         1
A              29         2 
B              12         1 
E              55         1

Answer 1

我建议创建一个没有行列的临时数据集。然后.drop_duplicates() 得到每个音符一行。然后调用.sample() 选择您的随机子集，然后调用.merge() 将样本重新加入到患者id 和note_id 上的原始数据集。由于我不是熊猫专家，因此可能有更快的方法。