Python mplfinance addplot 问题。添加入场信号

Question

有谁知道如何在我的图表上添加入场信号以便我可以直观地看到它？当 SPY 价格达到 411.50 美元时，我希望在图表上看到进场信号。

我已经阅读了我在网上可以找到的有关该主题的所有内容，包括 mplfinance 文档，似乎有一种方法可以通过使用 addplot() 散点图来做到这一点，但我不太明白。

到目前为止，我的代码如下:

# Libraries:
import pandas as pd
import mplfinance as mpf

# Pulling data:
data = pd.read_csv(r'C:\Users\Viktor\Documents\D2DT\Reformatted_Data\SPY_2021-04-12-2021-04-12.csv')

# Pre-processing data for mpf:
data['Date'] = pd.to_datetime(data['Date'])
data = data.set_index('Date')

# Plotting Data:
mpf.plot(data, type='candle', volume=True, style='yahoo', title="SP500 Data")

data.close()

CSV 内部: 感谢您对此的帮助!

Answer 1

@r-beginners，我弄乱了您在上一条评论中提供的链接中的文档并使其正常工作。谢谢。我发布了下面的代码，希望这对将来的人有帮助!

import pandas as pd
import numpy as np
import mplfinance as mpf

day = '2021-04-12'
data = pd.read_csv(r'C:\Users\Viktor\Documents\D2DT\Reformatted_Data\SPY_' + str(day) + "-" + str(day) + '.csv')

data['Date'] = pd.to_datetime(data['Date'])
data = data.set_index('Date')

signal = []

for i in data['close']:
    if i > 411.50:
        signal.append(i)
    else:
        signal.append(np.nan)

apd = mpf.make_addplot(signal, type='scatter', markersize=200, marker='v', color='r')

mpf.plot(data[day], type='candle', volume=True, style='yahoo', title="SP500 Data", addplot=apd)