Python:使用pyfolio中的plot_drawdown_underwater和pandas-datareader中的get_data_yahoo绘制单独的图时遇到了麻烦

Question

我有这样的代码：

from pandas_datareader import data as pdr
import pyfolio as pf

myStartDate = "2019-1-1"   # (Format: Year-Month-Day)
myEndDate = "2020-11-11"   # (Format: Year-Month-Day)
myTickers = ["AAPL", "MSFT"]

myData = pdr.get_data_yahoo(myTickers, myStartDate, myEndDate)

myDailyAdjustedClose = myData[['Adj Close']]
returns = myDailyAdjustedClose.pct_change()

pf.plot_drawdown_underwater(returns)

问题是，苹果和微软都是在同一张图上绘制的。相反，我想要两个独立的地块。

除了"pf.plot_drawdown_underwater(returns)“，我还尝试了：

for x in range(returns.shape[1]):
    columnData = returns.iloc[:, x]
    pf.plot_drawdown_underwater(columnData)

但我还是得到了同样的结果。

Answer 1

这是可行的：

from pandas_datareader import data as pdr
import pyfolio as pf
import matplotlib.pyplot as plt

myStartDate = "2019-1-1"   # (Format: Year-Month-Day)
myEndDate = "2020-11-11"   # (Format: Year-Month-Day)
myTickers = ["AAPL", "MSFT"]

myData = pdr.get_data_yahoo(myTickers, myStartDate, myEndDate)

myDailyAdjustedClose = myData[['Adj Close']]
returns = myDailyAdjustedClose.pct_change()

returnsColumnCount = returns.shape[1]

fig, ax = plt.subplots(returnsColumnCount, 1, figsize=(10, 10))
fig.tight_layout(pad=5)

for x in range(returnsColumnCount):
    columnData = returns.iloc[:, x]
    pf.plot_drawdown_underwater(columnData, ax=ax[x])

plt.show();