1000万用户标签存内存炸了?Python布尔数组从800MB到8MB的踩坑实录

📅 2026/8/9 1:07:58
1000万用户标签存内存炸了?Python布尔数组从800MB到8MB的踩坑实录
「部分情节为虚构演绎仅供参考」我所在的团队做的是推荐系统核心资产之一就是用户标签体系。千万级用户每个用户身上挂着几百个布尔标签有没有点击过运动品类、是不是高消费人群、最近7天有没有活跃、有没有领过优惠券……听起来很朴素对吧就是一堆True和False。但就是这一堆True和False差点把我给整「标炸」了——不是标签的「标」是爆炸的「炸」。内存炸、时间炸、心态也炸。从 list 到各种主流方案数据一涨全线 OOM最朴素的list[bool]一开始我们用的是最朴素的 Pythonlistuser_tags[False]*10_000_000# 1000万个用户每个一个布尔标记这行代码跑起来的那一刻我仿佛听到了服务器风扇的哀嚎。一个 Pythonlist里存的不是True/False本身而是指向 PyObject 的指针。每个指针 8 字节1000万个元素光指针就是 80MB。但这才一个标签啊我们有几百个标签80MB × 300 24GB直接把内存干穿。更离谱的是Python 的bool是int的子类True和False在内存里是两个全局单例对象。你往 list 里塞 1000万个True其实是在塞 1000万个指向同一个对象的指针。这就像你往仓库里堆了 1000万张写着「有这个标签」的纸条但标签本身只有一个。内存墙第一次亮起了红灯。array(b)省了内存慢了速度后来我们换成了array模块fromarrayimportarray user_tagsarray(b,[0])*10_000_000# 有符号 char1字节内存确实降下来了1000万个元素只要 10MB。但问题来了随机访问和修改的速度感人。array的每次索引访问都要做类型检查和装箱拆箱在千万级别的循环里这个开销被放大。跑一次全量标签扫描做用户圈选等得人想睡觉。内存墙是拆了时间墙又立起来了。numpy.ndarray快是真快但……numpy的bool_数组每个元素只占 1 字节底层是 C 连续内存向量化操作快得飞起importnumpyasnp user_tagsnp.zeros(10_000_000,dtypenp.bool_)内存 10MB速度也够快。看起来完美直到我们遇到动态更新的场景。用户标签是实时变化的——用户点了个商品、领了张券、取消了关注标签就要跟着变。而numpy的数组是定长的每次np.append都会创建新数组、拷贝全部数据。更致命的是当布尔数组极度稀疏比如「最近30天购买过奢侈品」这个标签1000万用户里可能只有10万人是True时numpy依然老老实实为每个元素分配 1 字节。99% 的空间都在存False纯纯的浪费。稀疏矩阵scipy.sparse稀疏的救星但不是布尔的家我们试过scipy.sparse稀疏场景下内存确实省了。但它是为数值矩阵设计的不是为布尔数组设计的。它存的是非零元素的坐标和值对于布尔数组来说这个「值」字段纯属冗余它的索引是int64每个坐标 8 字节对于千万规模光坐标就比numpy的 1 字节/元素还贵它的 API 是矩阵语义dot、matmul不是数组语义append、pop、find。用起来别扭不说性能也没占到便宜。小结主流方案全军覆没方案内存1000万 bool随机访问动态修改稀疏场景list[bool]~80MB快快浪费array(b)~10MB慢慢浪费numpy.ndarray~10MB快灾难浪费scipy.sparse看稀疏度慢慢语义错位四条路四条死胡同。内存墙、时间墙、语义墙三面夹击。破局思路给布尔数组装个「自动变速箱」一个普通人就知道的现象内存墙在讲方案之前先聊一个普通人就知道的现象内存占用多就卡。你手机 8GB 内存开 20 个 App 就开始杀后台你电脑 16GB 内存开 50 个 Chrome 标签页就开始风扇狂转。这不是玄学这是内存墙Memory Wall。CPU 的运算速度和内存的读写速度之间存在数量级的鸿沟。当数据量超过 CPU 缓存L1/L2/L3的容量CPU 就不得不频繁去主存取数据而主存的速度比缓存慢 100 倍以上。数据量再大连主存都放不下就得去磁盘Swap那速度直接掉到每秒几 MB——比 CPU 慢 100 万倍。这里必须澄清一个常见的误解时间和空间是完全不相同的两部分跟能量守恒没半点关系。省内存的真正意义不是「省」本身而是把数据从慢的存储层级挪到快的存储层级。这才是「省内存 变快」的真正原因——不是时空转换而是数据离 CPU 更近了。构想像电风扇一样自动换挡那几天我满脑子都是这个问题。有天晚上盯着家里的电风扇发呆突然灵光一闪电风扇为什么省电因为它会根据温度自动换挡——热了就开三档猛吹凉了就切一档慢慢转。布尔数组为什么不能这样数据密集的时候就开「三档」——用紧凑的连续存储跑得快数据稀疏的时候就切「一档」——只记特殊值的位置省内存数据分布变了就自动换挡——不过注意换挡只在两个时机发生创建数组时以及调用optimize()时。平时 insert、pop、赋值它都不会偷偷换挡。我越想越兴奋连夜画了个图高密度低密度喂进来的数据密度有多高三档位图紧凑存储一档只存特殊值下标自动换挡器对外统一API同事听完点了点头然后问了一句让我当场噎住的话「那……挡位切换的时机怎么定数据一直在变会不会一会儿三档一会儿一档来回抖」我张了张嘴憋了半天最后只能说「这个……我还没想好。」但当时的我哪管这些觉得「自动变速箱」这个点子简直天才当晚就撸起袖子开干。自己做做了十几天疼到怀疑人生这十几天基本是这样度过的第一天写了个能跑的数组类能用开心。第二天换挡阈值写死成 50%结果数据一波动就疯狂来回切性能比不切还差。第三天想加个「滞回区间」防抖动结果阈值判断和实际存储对不上数据直接错乱。第四天稀疏区用array(I)存索引结果索引越界不报错静默写错位置排查了一整天。第五天加了「批量赋值」接口结果把「按索引赋值」和「按值过滤」两个语义写串了全乱套。第六天写了「按位取反」结果取反后count(True)对不上——稀疏区取反后忘了把特殊值从True换成False。第七天想支持in运算符结果每次判断都全量扫描比list还慢。第八天统计 True 个数的方法数字忽大忽小——缓存了结果但数据一变缓存没失效。第九天自动换挡函数写出来了但换挡瞬间要重建整个内部结构数据一多直接卡死。第十天想支持pickle序列化内部结构太复杂存进去再读出来全乱了。第十一天写了「查找第一个 True 的位置」稀疏区返回的是索引表里的位置不是数组里的真实位置差了好几个量级。第十二天盯着 2000 多行代码发现还有一堆边界条件没处理心态彻底崩了。到了第十二天我才意识到自己犯了一个致命错误——我把换挡做成了「每次数据变化都可能触发」的高频动作。正确的做法应该是换挡只在创建时和调用optimize()时发生平时操作都待在当前挡位里。那一刻我明白了从零锤一个生产可用的混合布尔数组真不是一个人十几天的事。我决定发帖求助。转机发帖求助被一句话点醒我把踩坑经历整理了一下发到了技术社区标题是「1000万个布尔标签list 爆内存、numpy 爆拷贝、scipy 爆语义我该怎么办」评论区画风出奇地一致——所有人都在推荐同一个库。其中一条评论直接点醒了我「你那个『自动变速箱』构想bool-hybrid-array早就实现好了。而且它换挡只在两个时机发生创建时和调用optimize()时。平时 insert、pop、赋值都不换挡所以不会来回抖。你之前疯狂换挡是因为你把换挡时机搞错了——换挡是低频动作不是高频动作。」我盯着这条评论看了半天突然就通了。评论区其他声音也出奇地一致「别折腾了直接pip install bool-hybrid-array你这个问题它天生就是为这个设计的。」「我之前用 numpy 存 2 亿个布尔标记内存直接爆换它之后 1% 稀疏场景内存降了 90%……」「自己看它的memory_usage(detailTrue)输出数字不会骗人。」「密集区用 numpy、稀疏区用 array两边都是成熟方案不是野路子。」「它在 PyPI 上月下载过万GitHub 上迭代了很多版本不是课程作业。」「支持 numpy 直接转换np.array(arr)一行就接进现有 pipeline 了。」「MIT 协议商用随便用。」「我从 Python 3.9 到 3.14 全跑过PyPy 也没问题。」「它的find和rindex在稀疏区返回的是真实位置不是索引表位置。」说实话评论区清一色夸同一个库看着像水军。但我想通了是不是水军跟我没关系我只关心它在我机器上跑出来的数字是不是真的。所以我自己动手验。frombool_hybrid_arrayimportBoolHybridArr# 1000万个布尔标签只有1%是True比如「购买过奢侈品」这种稀疏标签tagsBoolHybridArr(i%1000foriinrange(10_000_000))print(tags.memory_usage(detailTrue))看到输出的数字我第一反应是「这库是不是在输出里造假」。所以我拿tracemalloc分别测了list[bool]、numpy和bool-hybrid-array三者的真实内存又用time.perf_counter()各跑了三遍取中位数。结果跟它memory_usage(detailTrue)报的数字对得上误差很小。数字不是我编的是它自己报的而且我验过。你要是也怀疑把代码复制到你机器上跑一遍就知道了。不过我得说句公道话memory_usage(detailTrue)的数字是它自己算的不是第三方审计的。我能保证的是我用tracemalloc独立测出来的结果跟它对得上。别信我也别信它信你自己的测量。