我刚刚了解了枚举,并认为它们将适合我正在编码的东西.但是当我运行这个代码时,我得到了一个错误.我是在做不该做的事情还是这是个错误?
当try 使用枚举来分组时,我会遇到以下错误:
TypeError: '<' not supported between instances of 'CarBrand' and 'CarBrand'
代码:
import pandas as pd
from enum import Enum
class CarBrand(Enum):
VOLVO = 'Volvo'
BMW = 'BMW'
data = {
'brand': [CarBrand.VOLVO,
CarBrand.VOLVO,
CarBrand.BMW],
'price': [35000,
37000,
45000]
}
df = pd.DataFrame(data)
sum_per_brand = df.groupby('brand').sum('price')
print(sum_per_brand)
这是我期待的输出: 品牌价格 宝马 45000 沃尔沃72000