我如何在字典中引用Pandas数据帧中的键？

Question

我有一本字典，里面放了几个数据帧（在这一点上都是一样的）。我试图将数据添加到每个数据帧（财政年度）的同一列中，该列对应于调用每个数据帧的键。我指定的关键是财政年度。当我尝试使用dict.项目（），但是，它为每个数据帧分配相同的值（上一个会计年度）。我的目标是预测每一个财政年度的收入，我将根据每一年将获得的收入分成一个新的列。我将代码简化为以下内容：

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])

Answer 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])