如何大熊貓數據幀選擇行基於類別

Question

這實在是微不足道，但無法相信我已經爲一個小時左右徘徊，仍然可以找到答案，所以在這裏你是：

df = pd.DataFrame({"cats":["a","b"], "vals":[1,2]}) 
    df.cats = df.cats.astype("category") 
    df 

我的問題是如何選擇其「貓」列的類別是「a」的行。我知道df.loc[df.cats == "a"]可以工作，但它基於元素上的平等。有沒有一種方法可以根據類別的級別進行選擇？

Answer 1

這工作：

df.cats[df.cats=='a'] 

UPDATE

的問題進行了更新。新解決方案：

df[df.cats.cat.categories == ['a']]