附加到熊猫数据框中的列表

Posted 2023-03-12

【中文标题】附加到熊猫数据框中的列表

【英文标题】：Appending to a list inside a pandas dataframe

【发布时间】：2020-05-21 19:24:55

【问题描述】：

我有一个带有一列空列表的 pandas 数据框。我想在数据框中的特定位置附加到此列表。下面是我正在使用的代码。

df = pd.DataFrame([[1, 2], [4, 5], [7, 8]],
                  index=['cobra', 'viper', 'sidewinder'],
                  columns=['max_speed', 'shield'])
df['new_col'] = [[]] * df.shape[0]
df.loc['viper', 'new_col'].append('a')

我期望和想要的输出是：

            max_speed  shield new_col
cobra               1       2   []
viper               4       5   [a]
sidewinder          7       8   []

但我得到的输出是：

            max_speed  shield new_col
cobra               1       2   [a]
viper               4       5   [a]
sidewinder          7       8   [a]

为什么会发生这种情况，如何获得预期的输出？谢谢

【参考方案1】：

@Joylove 在 cmets 中回答了这个问题here。

所有这些列表都是同一个对象。设置一个单元格将设置所有单元格。

使用df['new_col'] = [[] for _ in range(df.shape[0])] 可以解决此问题。