python - 我正在尝试使用 pandas.DataFrame.replace 替换包含具有单个值的列表的单元格？

Question

我有一列包含 6 个唯一值的数据框：

1, 2, 3, [1,2], [2,3], [1,2,3]

当我尝试以下代码时，我正在尝试使用 pandas.DataFrame.replace 将 [1,2] 替换为 4：

x = df2['lockdown_level'].replace([1,2],4)

我得到一个输出：

4, 4, 3, [1,2], [2,3], [1,2,3]

而我希望唯一值的输出为：

1, 2, 3, 4, [2,3], [1,2,3]

有没有人有任何建议用单个值替换列表？

标签： pythonpandaslistdataframe

Answer 1

我会这样做： df2['lockdown_level'].apply(lambda x: 4 if x==[1, 2] else x)

埃莱奥诺拉