用return语句替换print（）[重复]

Question

使用 return 语句后，我需要在数据框中输出我的输出。 print() 显示我的输出，但 return 语句给了我：{}

这是我当前的打印代码Entire Code with current output

这是我的输入数据框格式：

这是我的代码主体

prediction = {}
df2 = df.rename(columns={'date': 'ds','load': 'y', 'account': 'article'})
list_articles = df2.article.unique()

def get_prediction(df):
    prediction = {}
    f2 = df.rename(columns={'date': 'ds','load': 'y', 'account': 
    'article'})
    list_articles = df2.article.unique()  
 for article in list_articles:
     article_df = df2.loc[df2['article'] == article]
     my_model = Prophet(weekly_seasonality= True, 
     daily_seasonality=True,seasonality_prior_scale=1.0)
     my_model.fit(article_df)
     future_dates = my_model.make_future_dataframe(periods=6, freq='MS')
     forecast = my_model.predict(future_dates)
     prediction[article] = forecast
return prediction

我需要关于预测的 return 语句来执行打印显示的内容。目前使用“return prediction”返回的所有内容 = {} - 一个空字典

Answer 1

你需要修正你的缩进：

prediction = {}
df2 = df.rename(columns={'date': 'ds','load': 'y', 'account': 'article'})
list_articles = df2.article.unique()

def get_prediction(df):
    prediction = {}
    f2 = df.rename(columns={'date': 'ds','load': 'y', 'account': 
    'article'})
    list_articles = df2.article.unique()  
    for article in list_articles:
        article_df = df2.loc[df2['article'] == article]
        my_model = Prophet(weekly_seasonality= True, 
        daily_seasonality=True,seasonality_prior_scale=1.0)
        my_model.fit(article_df)
        future_dates = my_model.make_future_dataframe(periods=6, freq='MS')
        forecast = my_model.predict(future_dates)
        prediction[article] = forecast
    return prediction

for循环需要在函数下，for循环的内容需要在for循环下，return需要在函数下。