Pandas：如何從 dataframe 的特定列中獲取每個類別的樣本行並保存在單個 csv 中？

Question

下面是 dataframe (df)。 我想從“國家”列的每個類別中保存 3 行的樣本。 以下是我的代碼，但它不是基於類別保存的。 我需要單個 csv 有樣品。 請建議。

data = {'country':['India', 'Nepal', 'Canada', 'USA','India', 'Nepal', 'Canada', 'USA','India', 'Nepal', 'Canada', 'USA','India', 'Nepal', 'Canada', 'USA','India', 'Nepal', 'Canada', 'USA'],
    'Age':[20, 21, 19, 18,20, 21, 19, 18,20, 21, 19, 18,20, 21, 19, 18,20, 21, 19, 18]}
df = pd.DataFrame(data)

df.sample(n=3).to_csv(sampledata.csv, na_rep='NA', index = False)

Answer 1

GroupBy 然后采樣

df.groupby('country').sample(3)

   country  Age
2   Canada   19
6   Canada   19
10  Canada   19
4    India   20
0    India   20
12   India   20
1    Nepal   21
13   Nepal   21
9    Nepal   21
3      USA   18
11     USA   18
19     USA   18