如何将Pandas DataFrame中的NaN值替换为指定长度的零列表

报错原因 #

你遇到的赋值报错本质是Pandas的默认赋值逻辑导致的：当你向.loc选中的多行列赋值一个列表时，Pandas会尝试将列表的每个元素依次匹配到每一行，而不是将整个列表作为单个值填入每个单元格。只要你的零列表长度和空值行的数量不一致，就会触发长度不匹配的错误。

最优实现方案 #

方案1：仅处理空值行（效率最高，推荐） #

不需要遍历全量数据，仅对空值行做赋值，性能最优：

import numpy as np
import pandas as pd

# 你原有计算最大长度、生成零列表的逻辑保持不变
max_wid = dataframe["title"].map(lambda x: len(x)).max()
zeros = np.zeros(max_wid, dtype=int).tolist()

# 新增：统计空值行数量，生成等长的零列表集合做赋值
na_cnt = dataframe["title"].isna().sum()
# 如果后续需要修改填充的零列表且不影响其他行，用copy生成独立列表
dataframe.loc[dataframe.title.isnull(), "title"] = [zeros.copy() for _ in range(na_cnt)]
# 如果后续不会修改填充的零列表，可以简化为 [zeros]*na_cnt ，性能更高

方案2：apply实现（写法更简洁） #

适合数据量不大的场景，代码更短易读：

max_wid = dataframe["title"].map(lambda x: len(x)).max()
zeros = np.zeros(max_wid, dtype=int).tolist()

dataframe["title"] = dataframe["title"].apply(lambda x: x if x is not None else zeros.copy())

注意事项 #

如果你的title列中除了列表和NaN还存在其他合法非空值，需要对应调整apply中的判断条件，避免误替换。

内容的提问来源于stack exchange，提问作者Nicolò Antonucci