如何根據數據框中的列值獲取特定行數

Question

假設我有這種方式的MNIST數據集。

df = pd.read_csv('data/train.csv') 
data = df.loc[df['label'].isin([1,6])] 

我想只選擇那些行其列[「標籤」] == 1或6

不過，我想從每列僅獲得500行[「標籤」]

我該怎麼做？

Answer 1

使用GROUPBY第一，然後過濾器，即

ndf= df.groupby('label').head(500) 
data = ndf.loc[ndf['label'].isin([1,6])] 

Answer 2

您可以將它們分組，然後選擇要爲每個值數：

data = df.loc[df['label'].isin([1,6])].groupby('label').head(500)