登录
首页 >  文章 >  python教程

Python用户ID分类求和实战教程

时间:2026-04-14 14:54:44 161浏览 收藏

本文手把手教你用Python精准统计指定用户的充值总额(server_id为None)与消费总额(server_id非None),并自动计算最终余额,彻底规避循环中反复重置变量导致的求和错误;通过生成器+sum()的声明式写法,代码简洁、无副作用、可读性强,同时涵盖None判断规范、空值防护、大数据预分组优化及金融精度提醒等实战要点,助你一招解决游戏充值、SaaS账户管理等场景下的用户资金聚合难题。

Python 中基于用户 ID 对交易记录按资金流向分类求和的完整教程

本文详解如何使用 Python 高效统计指定用户的“充值总额”(server_id 为 None)与“消费总额”(server_id 非 None),并计算最终余额,避免常见循环重置变量导致的逻辑错误。

本文详解如何使用 Python 高效统计指定用户的“充值总额”(server_id 为 None)与“消费总额”(server_id 非 None),并计算最终余额,避免常见循环重置变量导致的逻辑错误。

在处理用户资金流水类业务时(如游戏服务器充值/扣费、SaaS 账户余额管理),常需从原始交易列表中按用户 ID 聚合两类操作:

  • 充值(add_funds):当 server_id is None,表示该笔金额为用户主动添加至账户的信用额度;
  • 消费(spent):当 server_id is not None,表示用户在某服务器上实际支出的费用。

目标是:对 list2 中每个 id,精准汇总其在 list1 中所有匹配交易的充值总和与消费总和,并输出 final_balance = add_funds - spent。

✅ 正确实现思路(关键:分离聚合逻辑,避免状态污染)

原代码的核心缺陷在于:

  • 在内层循环中反复初始化 spent = 0 和 add_fund = 0,导致每次匹配单条记录就重置计数器;
  • 未对同一用户的多条交易做累积求和,而是逐条覆盖;
  • 缺少用户级数据结构存储中间结果,无法支撑最终格式化输出。

推荐解法:使用生成器表达式 + sum() 进行声明式聚合,语义清晰、无副作用、性能良好:

list1 = [
    {'user_id': 1, 'server_id': '10', 'amount': 100},
    {'user_id': 2, 'server_id': None, 'amount': 10000},
    {'user_id': 3, 'server_id': '10', 'amount': 200},
    {'user_id': 1, 'server_id': '10', 'amount': 200},
    {'user_id': 2, 'server_id': '10', 'amount': 110},
    {'user_id': 2, 'server_id': None, 'amount': 40000},
    {'user_id': 3, 'server_id': '10', 'amount': 100},
    {'user_id': 3, 'server_id': None, 'amount': 12500},
    {'user_id': 3, 'server_id': '10', 'amount': 100},
    {'user_id': 1, 'server_id': None, 'amount': 22500},
]

list2 = [
    {'id': 1},
    {'id': 4},
    {'id': 18},
    {'id': 3},
    {'id': 2},
]

# 存储每位目标用户的汇总结果
user_balances = {}

for user in list2:
    user_id = user['id']

    # ✅ 精准筛选:同一 user_id 且 server_id 为 None 的所有 amount 并求和
    add_funds = sum(
        item['amount'] for item in list1 
        if item['user_id'] == user_id and item['server_id'] is None
    )

    # ✅ 精准筛选:同一 user_id 且 server_id 非 None 的所有 amount 并求和
    spent = sum(
        item['amount'] for item in list1 
        if item['user_id'] == user_id and item['server_id'] is not None
    )

    final_balance = add_funds - spent

    # 仅当该用户有任一类型交易时才记录(可选过滤)
    if add_funds != 0 or spent != 0:
        user_balances[user_id] = {
            'spent': spent,
            'add_funds': add_funds,
            'final_balance': final_balance
        }

# 格式化输出(按 list2 顺序,含千位分隔符增强可读性)
for user in list2:
    uid = user['id']
    if uid in user_balances:
        b = user_balances[uid]
        print(f"User {uid} spent {b['spent']:,}, added funds {b['add_funds']:,}, final balance {b['final_balance']:,}")
    else:
        print(f"User {uid} has no transaction records.")

运行输出:

User 1 spent 300, added funds 22,500, final balance 22,200
User 4 has no transaction records.
User 18 has no transaction records.
User 3 spent 400, added funds 12,500, final balance 12,100
User 2 spent 110, added funds 50,000, final balance 49,890

⚠️ 注意事项与最佳实践

  • is None vs == None:始终使用 is None 判断 None 值,这是 Python 官方推荐写法,语义明确且性能更优;
  • 空值安全:若 list1 中存在缺失键(如无 'server_id' 字段),建议用 item.get('server_id') 替代直接索引,避免 KeyError;
  • 扩展性优化:若数据量大(>10k 条),可先将 list1 按 user_id 预分组为字典({user_id: [tx1, tx2, ...]}),避免每次重复遍历全量列表;
  • 精度保障:金融场景中,金额应使用 decimal.Decimal 类型而非 float,防止浮点误差(本例因均为整数可暂不启用);
  • 输出一致性:list2 中用户 4 和 18 在 list1 中无对应记录,程序正确跳过——这体现了健壮性设计,而非隐式忽略。

通过本方案,你不仅能获得预期结果,更掌握了「以用户为中心聚合多维条件数据」的通用模式,适用于日志分析、账务对账、用户行为统计等广泛场景。

文中关于的知识介绍,希望对你的学习有所帮助!若是受益匪浅,那就动动鼠标收藏这篇《Python用户ID分类求和实战教程》文章吧,也可关注golang学习网公众号了解相关技术文章。

资料下载
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>