如何将Pandas日期Series转换为mplfinance绘图的vlines参数

问题场景 #

需要将Pandas中的日期Series转换为mplfinance绘图函数vlines参数可接受的格式。测试代码如下：

import numpy as np
import pandas as pd

df = pd.DataFrame({'dt': ['2021-2-13', '2022-2-15'],
                   'w': [5, 7],
                   'n': [11, 8]})
df.reset_index()
print(list(df.loc[:,'dt'].values))

输出为：['2021-2-13', '2022-2-15']，而mplfinance的vlines参数需要类似[('2021-2-13'), ('2022-2-15')]的列表格式。

用户尝试直接赋值.values给my_vlines无效：

import mplfinance as mpf

RES['Date'] = RES['Date'].dt.strftime('%Y-%m-%d')
my_vlines=RES.loc[:,'Date'].values  # 无法正常工作

fig, axlist = mpf.plot( ohlc_df, type="candle",  vlines= my_vlines, xrotation=30, returnfig=True, figsize=(6,4))

解决方案 #

将.values转换为列表格式即可，代码如下：

my_vlines=list(RES.loc[:,'Date'].values) 

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