【跟着AI学Python】Python进阶:掌握 cProfile + pstats 进行函数级性能分析

# 分析整个脚本，按累计时间排序python -m cProfile -s cumulative your_script.py# 保存结果到文件（供后续分析）python -m cProfile -o profile_output.prof your_script.py

import cProfiledef main():    # 你的业务逻辑    process_data()# 仅分析 main() 函数if __name__ == "__main__":    cProfile.run('main()', 'profile_output.prof')

import cProfilewith cProfile.Profile() as pr:    # 只分析这段代码    for i in range(1000):        expensive_function(i)# 保存或打印结果pr.dump_stats('loop.prof')pr.print_stats(sort='cumulative')

# 启动 pstats 交互环境python -m pstats profile_output.prof

# analyze_profile.pyimport pstatsfrom pstats import SortKey# 加载性能数据stats = pstats.Stats('profile_output.prof')# 按累计时间排序，显示前 15 行stats.sort_stats(SortKey.CUMULATIVE).print_stats(15)# 过滤：只显示 bank.py 中的函数stats.print_stats('bank.py')# 统计特定函数的调用者stats.print_callers('deposit')

import timeimport randomclass BankAccount:    def __init__(self, owner):        self.owner = owner        self.balance = 0.0    def deposit(self, amount):        # 模拟风控检查（实际可能更复杂）        self._validate_amount(amount)        self._log_transaction("deposit", amount)        self.balance += amount    def _validate_amount(self, amount):        if amount <= 0:            raise ValueError("Invalid amount")        time.sleep(0.0001)  # 模拟 I/O    def _log_transaction(self, action, amount):        # 模拟写日志        with open("transactions.log", "a") as f:  # ← 瓶颈！            f.write(f"{action}: {amount}\n")def simulate_transactions():    accounts = [BankAccount(f"User{i}") for i in range(50)]    for _ in range(200):  # 10,000 笔交易        acc = random.choice(accounts)        acc.deposit(random.uniform(10, 100))if __name__ == "__main__":    simulate_transactions()

python -m cProfile -o bank.prof bank_system.py

# analyze_bank.pyimport pstatsstats = pstats.Stats('bank.prof')stats.sort_stats('cumulative').print_stats(10)

pip install pyprof2calltree# window下 安装 snakeviz（Web 界面，跨平台）pip install snakeviz

pyprof2calltree -i bank.prof -k  # 直接启动 qcachegrind# 或生成 callgrind 文件pyprof2calltree -i bank.prof -o callgrind.out

# 使用 snakeviz 打开 .prof 文件snakeviz bank.prof