Python股票收益分析代码缺失参数补全及plt.legend报错求解

错误1：comments.append方法缺失参数修复 #

原代码字符串使用位置占位符{0}、{1}、{2}，按顺序对应三个输入参数：

• {0}：对数累计收益率final_cum_log_return
• {1}：换算后的百分比简单收益率exp_final_cum_log_return
• {2}：当前股票代码ticker
  你自行补全的代码错误原因是传入了整个股票列表tickers和完整收益率DataFramereturns，而非当前循环的单个股票对应的值。
  正确补全后的代码：

comments.append("{2}: Cum log return = {0}.\nSimple return = (exp({0}) -1)*100= {1} %. \n".format(final_cum_log_return, exp_final_cum_log_return, ticker))

错误2：No handles with labels found to put in legend报错修复 #

该报错的核心原因是原代码存在笔误：

# 错误写法：点号会把ax变成fig的属性，实际没有成功赋值子图对象
fig.ax = plt.subplots(figsize=(20,5))
# 正确写法：用逗号拆分返回的画布和子图对象
fig, ax = plt.subplots(figsize=(20,5))

赋值错误导致ax.plot没有在实际绘制的子图上添加带label的曲线，plt.legend找不到对应标签元素就会报错。

额外优化建议 #

将收益率计算逻辑移到循环外，避免每次循环重复计算所有股票的收益率，提升运行效率，完整修正代码如下：

price = data["Adj Close"].loc["2020-03-02":"2020-12-21"]
returns = price.apply(lambda x:np.log(x / x.shift(1))).dropna()
fig, ax = plt.subplots(figsize=(20,5))
comments = []
for ticker in tickers:
    title="{} . cumulative log return".format(ticker)
    ax.plot(returns.index,returns[ticker].cumsum(), label=ticker)
    final_cum_log_return = returns[ticker].cumsum().iloc[-1]
    exp_final_cum_log_return = (np.exp(final_cum_log_return)-1)*100
    comments.append("{2}: Cum log return = {0}.\nSimple return = (exp({0}) -1)*100= {1} %. \n".format(final_cum_log_return, exp_final_cum_log_return, ticker))
plt.legend()
plt.show()
print("On Dec. 21:\n")
[print(x) for x in comments]

内容的提问来源于stack exchange，提问作者user16233760