Pythonの特定のクラスから行数をサンプリングするには？

Question

「ラベル」列のクラス= 1の「のみ」から2行をサンプルしたいとします。あなたがいることがわかります私のコードで

：

1）私は、クラス= 1（4行）

2）からすべての行をサンプリングそれから私は、以前のデータフレーム

から2行をサンプリングしかし、私はこれを行うより良い方法がなければならないと確信しています。

# Creation of the dataframe 
df = pd.DataFrame(np.random.rand(12, 5)) 
label=np.array([1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3]) 
df['label'] = label 


# Sampling 
df1=df.loc[df['label'] == 1] #Extract ALL samples with class=1 
df2 = pd.concat(g.sample(2) for idx, g in df1.groupby('label')) #Extract 2 samples from df1 
df2 

Answer 1

私はこれを行うだけだろう：

df1.query('label == 1').sample(2)