如何为熊猫列中的每个列表附加不同的元素？

Question

我有一个数据框 df，其中包含特定列 col_a 中的列表。例如，

df = pd.DataFrame()
df['col_a'] = [[1,2,3], [3,4], [5,6,7]]

我想在这些列表上使用条件并应用特定的修改，包括追加。例如，假设如果列表的长度> 2，我想追加另一个元素，这是当前列表的最后两个元素的总和。所以，考虑到上面的第一个列表，我有 [1, 2, 3] 并且我想要 [1, 2, 3, 5]。

我尝试做的是：

df.loc[:, col_a] = df[col_a].apply(
                    lambda value: value.append(value[-2]+value[-1]) 
                          if len(value) > 1 else value)

但是对于该列的所有元素，该列中的结果是None。

有人可以帮帮我吗？

非常感谢您。

Answer 1

问题在于append 是一个就地函数并返回None。您需要将两个列表添加在一起。所以一个带有虚拟变量的工作示例是：

df = pd.DataFrame({'cola':[[1,2],[2,3,4]], 'dum':[1,2]})
df['cola']=df.cola.apply(lambda x: (x+[sum(x[-2:])] if len(x)>2 else x))

Answer 2

如果你想使用 append 试试这个：

def my_logic_for_list(values):
    if len(values) > 2:
        return values + [values[-2]+values[-1]]
    return values
df['new_a'] = df['a'].apply(my_logic_for_list)

你不能在 lambda 函数中使用 append。