如何编写可复用Python函数计算两年营收相对差值？

先修正你代码里的错误，再封装成可复用的函数：

核心修正与函数封装 #

import pandas as pd

def calculate_revenue_diff(df, target_year, base_year):
    # 复制原数据，避免修改原始DataFrame
    df_processed = df.copy()
    # 提取年份（注意你的日期是日-月-年格式，要加dayfirst=True）
    df_processed['year'] = pd.DatetimeIndex(df_processed['date'], dayfirst=True).year
    # 计算单条订单营收（原代码错误：order.quantity改为df_processed.quantity）
    df_processed['revenue'] = df_processed['price'] * df_processed['quantity']
    # 按年份聚合总营收
    yearly_revenue = df_processed.groupby('year')['revenue'].sum()
    
    # 检查指定年份是否存在
    if target_year not in yearly_revenue.index or base_year not in yearly_revenue.index:
        raise ValueError("指定的年份在数据中没有对应营收记录")
    
    # 计算相对差值：(目标年营收 / 基准年营收) - 1
    return (yearly_revenue[target_year] / yearly_revenue[base_year]) - 1

使用示例 #

假设你的订单数据存储在df中，调用方式如下：

# 构造示例数据
data = {
    'order_id': [1,2,3,4,5,6],
    'price': [30,50,45,200,54,200],
    'quantity': [2,4,3,4,2,4],
    'date': ['20-03-2022','21-04-2020','20-03-2020','21-04-2019','04-03-2020','15-04-2019']
}
df = pd.DataFrame(data)

# 计算2019年相对2020年的营收差值
diff = calculate_revenue_diff(df, 2019, 2020)
print(diff)

原代码错误说明 #

• df.price*order.quantity：这里order未定义，应该引用DataFrame的quantity列，改为df['price'] * df['quantity']
• total_revenue.loc[total_revenue["year"] = 2019 ]：groupby后的total_revenue是Series，索引就是年份，直接用total_revenue[2019]即可，且条件判断要用==而非=
• 日期处理未加dayfirst=True：你的日期格式是日-月-年，不指定会导致年份识别错误

内容的提问来源于stack exchange，提问作者Pixel