Python Pandas修改DataFrame单元格列表值后数据异常的解决咨询

问题场景 #

尝试用Pandas提取DataFrame指定单元格中的列表，修改列表内元素后放回原单元格，但执行后所有行的列表都变成了相同值。

原代码 #

index_a = 1

temp_list = []
# 提取index_a行mycolumn列的列表
temp_list = df.loc[index_a,  'mycolumn']
# 将列表第5个元素（索引4）加上mynumber
temp_list[4] = temp_list[4] + mynumber
# 将修改后的列表放回原单元格
df.at[index_a,  'mycolumn'] = temp_list

temp_list = []
# 提取index_a行mynextcolumn列的列表
temp_list = df.loc[index_a,  'mynextcolumn']
# 将列表第5个元素（索引4）加上myothernumber
temp_list[4] = temp_list[4] + myothernumber
# 将修改后的列表放回原单元格
df.at[index_a,  'mynextcolumn'] = temp_list

问题现象 #

执行后所有行的列表值完全一致：

index    mycolumn    mynextcolumn
  0    [0,4,0,0,6]   [0,4,0,0,6]
  1    [0,4,0,0,6]   [0,4,0,0,6]
  2    [0,4,0,0,6]   [0,4,0,0,6]

预期结果 #

仅修改指定行的列表，其他行保持原有值：

index    mycolumn    mynextcolumn
  0    [1,2,3,4,5]   [6,7,8,9,0]
  1    [0,0,1,1,2]   [0,4,0,0,6]
  2    [0,1,2,3,4]   [5,6,7,8,9]

原因分析 #

列表是可变对象，当你用df.loc[index_a, 'mycolumn']提取列表时，得到的是原列表的引用，而非副本。修改temp_list时，实际上直接修改了DataFrame中存储的原列表对象。如果DataFrame中多个行的单元格指向同一个列表对象（比如初始化时用了同一个列表赋值），就会导致所有行的列表同步变化。

解决方案 #

提取列表时必须创建列表的副本，修改副本后再放回原单元格，避免影响原列表对象。常用复制方式有两种：

1. 使用列表的copy()方法
2. 使用切片[:]创建副本

修改后的代码 #

index_a = 1

# 处理mycolumn列
# 提取列表副本
temp_list = df.loc[index_a, 'mycolumn'].copy()
temp_list[4] = temp_list[4] + mynumber
df.at[index_a, 'mycolumn'] = temp_list

# 处理mynextcolumn列
temp_list = df.loc[index_a, 'mynextcolumn'].copy()
temp_list[4] = temp_list[4] + myothernumber
df.at[index_a, 'mynextcolumn'] = temp_list

或者用切片方式实现：

index_a = 1

# 处理mycolumn列
temp_list = df.loc[index_a, 'mycolumn'][:]
temp_list[4] = temp_list[4] + mynumber
df.at[index_a, 'mycolumn'] = temp_list

# 处理mynextcolumn列
temp_list = df.loc[index_a, 'mynextcolumn'][:]
temp_list[4] = temp_list[4] + myothernumber
df.at[index_a, 'mynextcolumn'] = temp_list

额外注意事项 #

如果你的DataFrame是通过重复同一个列表初始化的（比如df = pd.DataFrame({'mycolumn': [initial_list]*3})），需要先确保每行的列表都是独立对象，否则即使修改副本放回，其他行还是会指向原列表。可以用列表推导式重新初始化：df = pd.DataFrame({'mycolumn': [initial_list.copy() for _ in range(3)]})

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