計算列表中的類型對象

Question

你有一個很好的方法來計算我的goblet的Sugar數嗎？

class Sugar:pass
class Milk:pass
class Coffee:pass

goblet = [Sugar(), Sugar(), Milk(), Coffee()]
sugar_dosage = goblet.count(Sugar) #Doesn't work

Answer 1

您可以使用sum和isinstance的生成器表達式 ：

sum(isinstance(x, Sugar) for x in goblet)
# or
sum(1 for x in goblet if isinstance(x, Sugar))

此外， list.count不起作用，因為該方法測試列表中有多少項等於 Sugar 。 換句話說，它基本上是為列表中的每個項目執行item==Sugar 。 這是不正確的，因為您想要測試每個項目的類型 ，這就是isinstance所做的。