能否补充商品品类字段

商品品类字段补充示例：

# 原数据（仅有商品名）
df_refund = pd.DataFrame({
    '退款单号': ['R-001', 'R-002'],
    '商品名': ['冰鲜三文鱼_20250301', '冷冻虾仁_20250302'],
    '金额': [50, 50],
    '时间': ['10:00', '10:01']
})

# 提取品类（商品名中关键词）
def extract_category(product_name):
    if '三文鱼' in product_name:
        return '冰鲜鱼类'
    elif '虾仁' in product_name:
        return '冷冻海鲜'
    else:
        return '其他'

df_refund['商品品类'] = df_refund['商品名'].apply(extract_category)

# 提取批次号
df_refund['批次'] = df_refund['商品名'].str.extract(r'(d{8})')

print(df_refund[['退款单号', '商品品类', '批次']])

输出：

注意：品类规则需根据实际商品名调整，建议建立品类映射字典（如 {'三文鱼':'冰鲜','虾仁':'冷冻'}）