首页 > 解决方案 > 随机对 pandas 列中的值求和并绘制它们

问题描述

我有一个数据集可以预测 300 只股票的回报。每行是一个观察值,如果预测它有负或正回报,则响应为 0 或 1,以及“回报”列中的实际百分比回报。我想从这 300 个预测中选择 50 个 300 次,并绘制出回报是多少。我到目前为止

df_p = pd.read_csv("portfolio.csv")

list = []
for i in 300:
    df_portfolio = df_p.sample(n=50)
    return = df_p['return'].sum()
    list.append(return)

plt.plot(list[:])
plt.show()

但我得到错误。我在 Python 中不是那么好,所以会很感激帮助

标签: python

解决方案


您不应该使用 list 或 return 作为变量名,因为它们都是特殊的 python 关键字。试试下面的:

df_p = pd.read_csv("portfolio.csv")

results_list = []
for i in 300:
    df_portfolio = df_p.sample(n=50)
    results_list.append(df_portfolio['return'].sum())
    

plt.plot(results_list)
plt.show()

推荐阅读