Добавление к списку внутри фрейма данных pandas - PullRequest
1 голос
/ 05 февраля 2020

У меня есть pandas фрейм данных со столбцом пустых списков. Я хочу добавить к этому списку в указанном c месте в кадре данных. Ниже приведен код, который я использую.

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]

Почему это происходит и как я могу получить ожидаемый результат? Thx

1 Ответ

1 голос
/ 06 февраля 2020

@ Joylove ответил на это в комментариях здесь .

Все эти списки являются одним и тем же объектом. Установка одной ячейки установит их все.

Использование df['new_col'] = [[] for _ in range(df.shape[0])] исправит это.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...