Как я могу ссылаться на ключ в фреймах данных Pandas в этом словаре?
У меня есть словарь, в который я поместил несколько фреймов данных (на данный момент все идентичны). Я пытаюсь добавить данные в один и тот же столбец каждого из этих фреймов данных (финансовый год) в соответствии с ключом, по которому может быть вызван каждый фрейм данных. Ключи, которые я назначил, - это финансовые годы. Однако, когда я пытаюсь использовать dict.items (), он присваивает каждому фрейму данных одно и то же значение (последний финансовый год). Цель состоит в том, чтобы спрогнозировать доход по финансовым годам, при этом я разделю выручку на новую колонку в зависимости от того, сколько будет получено за каждый год. Я упростил свой код до следующего:
import pandas as pd
columns = ['ID','Revenue','Fiscal Year']
ID = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
Revenue = [1000, 1200, 1300, 100 ,500, 0, 800, 950, 4321, 800]
FY = []
d = {'ID': ID, 'Revenue': Revenue}
df = pd.DataFrame(d)
df['Fiscal Year'] = ''
dataframe_dict = {}
def df_dict_func(start, end, dataframe):
date_range = range(start, end + 1)
for n in date_range:
dataframe_dict[n] = dataframe
for key, value in dataframe_dict.items():
value['Fiscal Year'] = key
df_dict_func(2018, 2025, df)
print(dataframe_dict[2019])
Ответы 1
Кажется ненужным создавать ключи и значения dict в одном цикле, а затем добавлять имя столбца в другом цикле. Вместо этого ваш код должен выглядеть примерно так
import pandas as pd
def df_dict_func(start, end, dataframe):
date_range = range(start, end + 1)
dataframe_dict = {}
for n in date_range:
sub = dataframe.copy()
sub['Fiscal Year'] = n
dataframe_dict[n] = sub
return dataframe_dict
columns = ['ID','Revenue','Fiscal Year']
ID = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
Revenue = [1000, 1200, 1300, 100 ,500, 0, 800, 950, 4321, 800]
FY = []
d = {'ID': ID, 'Revenue': Revenue}
df = pd.DataFrame(d)
df_dict = df_dict_func(2018, 2025, df)
print(df_dict[2019])
@DJHeels, Рад помочь!
Каков ваш ожидаемый результат?