Python：如何在表格存储变长列表？Pandas赋值报错求方案

我懂你在开发优化程序时遇到的这个困扰——想给pandas表格的单元格存不同长度的width离散值，但直接用table.loc[1,'Width'] = [100, 125, 150]报错对吧？这主要是因为pandas默认的列类型可能没设置对，或者赋值方式有小问题，下面给你几个实用的解决办法：

方案1：提前把列设为object类型 #

pandas的object类型可以存储任意Python对象（包括列表），你可以在创建表格时就指定好类型，或者把现有列转成这个类型：

import pandas as pd

# 方式1：创建表格时直接指定Width列为object
df = pd.DataFrame(
    {'Item': [1, 2, 3], 'Width': [[], [], []]},
    dtype={'Width': object},
    index=range(1,4)  # 设置1-based索引，符合你习惯的编号方式
)
# 现在赋值列表就没问题了
df.loc[1, 'Width'] = [100, 125, 150]
df.loc[2, 'Width'] = [100]
df.loc[3, 'Width'] = [100, 125]

# 方式2：给已有的表格转换列类型
# 如果你的表格已经存在，先把Width列转成object
df['Width'] = df['Width'].astype(object)
df.loc[1, 'Width'] = [100, 125, 150]

方案2：用at替代loc避免赋值问题 #

有时候报错可能是因为loc的链式索引触发了SettingWithCopyWarning，改用at直接定位单元格赋值会更稳妥：

# 先确保Width列是object类型
df['Width'] = df['Width'].astype(object)
# 用at精准赋值单个单元格
df.at[1, 'Width'] = [100, 125, 150]
df.at[2, 'Width'] = [100]
df.at[3, 'Width'] = [100, 125]

数据量较大时，也可以用apply批量设置值：

width_mapping = {1: [100,125,150], 2: [100], 3: [100,125]}
df['Width'] = df['Item'].apply(lambda x: width_mapping.get(x, []))

方案3：把列表展开为多行（更适合后续排列计算） #

既然你最终要生成所有排列组合，把嵌套列表展开成多行的方案会更实用——后续用itertools生成排列时会方便很多：

# 先构造带列表的DataFrame（1-based索引）
df = pd.DataFrame({
    'Item': [1,2,3],
    'Width': [[100,125,150], [100], [100,125]]
}, index=range(1,4))
# 把每个列表的元素展开成单独行
expanded_df = df.explode('Width', ignore_index=True)

展开后的数据表结构如下：

之后生成所有排列组合就很简单了：

import itertools

# 按Item分组获取每个Item的width选项
width_options = [group['Width'].tolist() for _, group in expanded_df.groupby('Item')]
# 生成所有可能的排列
all_permutations = list(itertools.product(*width_options))

这样得到的all_permutations就是你需要的所有width组合啦。

内容的提问来源于stack exchange，提问作者steve