如何将全局变量安全重构为局部状态变量

本文介绍在 Pandas apply() 场景下，如何消除 global 语句依赖，通过可变容器（如字典）或面向对象方式封装跨行计算所需的状态变量，提升代码可测试性、可重入性与线程安全性。

本文介绍在 pandas `apply()` 场景下，如何消除 `global` 语句依赖，通过可变容器（如字典）或面向对象方式封装跨行计算所需的状态变量，提升代码可测试性、可重入性与线程安全性。

在电信计费逻辑中，需按日期累计“已用优惠分钟数”并动态分配每通电话的优惠/标准资费分钟——这类跨行状态依赖常被误用全局变量实现。但 global 不仅破坏函数纯度，更导致难以单元测试、无法并发调用、状态污染风险高（例如多次运行结果相互干扰）。以下是两种专业、可落地的重构方案：

✅ 方案一：使用可变字典传参（推荐初阶重构）

将 prev_date 和 expen_minutes_used 封装进一个字典，作为显式参数传入函数。因字典是可变对象，函数内部可直接更新其键值，无需 global 声明：

from math import ceil

# 初始化状态容器（作用域清晰，生命周期可控）
state = {'prev_date': None, 'expen_minutes_used': 0}

def calculate_call_cost(row, state):
    # 日期变更时重置计数器
    current_date = row['data of call'].date()
    if state['prev_date'] is None or current_date != state['prev_date']:
        state['prev_date'] = current_date
        state['expen_minutes_used'] = 0

    # 计算本次可享优惠分钟数（最多补足至5分钟）
    duration_min = ceil(row['Duration in second'] / 60)
    expen_minutes = 0
    if state['expen_minutes_used'] < 5:
        expen_minutes = min(5 - state['expen_minutes_used'], duration_min)
        state['expen_minutes_used'] += expen_minutes

    # 剩余分钟按标准资费计费
    cheap_minutes = duration_min - expen_minutes
    return cheap_minutes * 0.40 + expen_minutes * 3.95

# 调用时显式传入状态字典（lambda 封装简洁）
calls_data_sorted['Cost of tariff'] = calls_data_sorted.apply(
    lambda row: calculate_call_cost(row, state), axis=1
)
total_cost2 = calls_data_sorted['Cost of tariff'].sum()

⚠️ 注意：该字典必须在 apply() 调用前初始化，且不能在循环中重复创建，否则状态无法累积。

✅ 方案二：封装为类（推荐中高阶场景）

当状态逻辑复杂或需复用时，定义 CallCostCalculator 类更符合工程规范。状态成为实例属性，方法天然共享上下文，且支持多实例隔离（如并行处理不同用户数据）：

from math import ceil

class CallCostCalculator:
    def __init__(self):
        self.prev_date = None
        self.expen_minutes_used = 0

    def calculate(self, row):
        current_date = row['data of call'].date()
        if self.prev_date is None or current_date != self.prev_date:
            self.prev_date = current_date
            self.expen_minutes_used = 0

        duration_min = ceil(row['Duration in second'] / 60)
        expen_minutes = 0
        if self.expen_minutes_used < 5:
            expen_minutes = min(5 - self.expen_minutes_used, duration_min)
            self.expen_minutes_used += expen_minutes

        cheap_minutes = duration_min - expen_minutes
        return cheap_minutes * 0.40 + expen_minutes * 3.95

# 实例化并应用（状态完全隔离）
calculator = CallCostCalculator()
calls_data_sorted['Cost of tariff'] = calls_data_sorted.apply(
    lambda row: calculator.calculate(row), axis=1
)
total_cost2 = calls_data_sorted['Cost of tariff'].sum()

? 关键总结

• 避免 global：它使函数行为依赖外部隐式状态，违背单一职责与可预测性原则；
• 优先选择类封装：状态管理清晰、可扩展性强、天然支持多实例；
• 字典方案适用轻量场景：改造成本低，适合快速迁移遗留代码；
• 切勿在 apply() 内部初始化状态：会导致每行重置，失去累计意义；
• 如需并发安全：类方案可配合 threading.Lock 或改用 concurrent.futures 分片处理。

通过以上重构，代码从“隐式状态依赖”升级为“显式状态传递”，不仅消除了全局变量缺陷，更为后续添加日志、调试钩子、策略切换等扩展能力奠定坚实基础。