Изменение и сохранение фрейма данных в словаре в цикле
Я работаю с кадром данных в цикле. На каждой итерации выполняются операции над переменными фрейма данных. В конце каждой итерации мне нужно сохранить этот фрейм данных в словаре с индексом, связанным с индексом итерации.
Например:
df = pd.DataFrame(index=range(20))
dict = {}
for k in range(5):
df['iter'] = k
dict[k] = df
Моим ожидаемым результатом 'dict' будет словарь с 5 фреймами данных. Скажем, для значения ключа «1» у меня должен быть фрейм данных «df» со столбцом «iter», в котором все значения равны 1. Точно так же для значения ключа «2» у меня должен быть «df» со всеми значениями 2.
Однако я обнаружил, что все ключевые значения имеют один и тот же фрейм данных, хранящийся в них. Все они имеют значение 4 в фрейме данных.
Я пробовал выполнять операции шаг за шагом, а не зацикливаться. Я обнаружил, что изначально хранится правильный фрейм данных. Но на следующем шаге итерации при выполнении
df['iter'] = k
значение в словаре также обновляется.
Как обойти эту проблему? Мой фактический фрейм данных намного больше и имеет гораздо больше операций, которые необходимо выполнять в цикле.
Ответы 2
Каждая запись в dict (ужасное имя, кстати, так как это уже название типа) должна быть копировать из df.
Вам нужно сделать копию фрейма данных. (dict — ужасное имя, не используйте ключевые слова в качестве имен переменных. Если вам нужно их использовать, ставьте после них знак подчеркивания.)
df = pd.DataFrame(index=range(20))
dict_ = {}
for k in range(5):
df['iter'] = k
dict_[k] = df.copy()