python - mplfinance 中是否有 plt.scatter 的等价物？如何在mplfinance中绘制数据点？

Question

mplfinance 中 plt.scatter 的等价物是什么？？？
我正在使用 mpl Finance 绘制股票价格图表。

def graph():
    file = 'prices1.xlsx'
    data = pd.read_excel(file, sheet_name = stockQuote)
    data.Date = pd.to_datetime(data.Date)
    data = data.set_index('Date')
    mpf.plot(data, type = 'candle', mav = (100), tight_layout = True)

这个

graph('AAPL')

应该让我得到烛台中的 AAPL 价格图表。
我有另一个带有买入和卖出价格的 excel 表。它看起来像这样

myPrices = pd.read_excel('transactions.xlsx')

日期
象征
行动
价钱


2020-03-20
AAPL
买
80

2021-03-05
AAPL
卖
120

2020-03-20
特斯拉
买
400


我知道 matplotlib 有这个:

plt.scatter(myPrices.index, myPrices['Buy'], label = 'Buy', market = '^', color = 'green')
plt.scatter(myPrices.index, myPrices['Sell'], label = 'Sell', market = 'v', color = 'red')

因为我想绘制'AAPL'，我想在transaction.xlsx 中读取日期，其中Symbol = 'AAPL'。我想在买入时在 MPLFINANCE GRAPH 中用绿色箭头 ^ 表示，在卖出时用红色箭头 v 表示。但是，我只知道matplotlib中的这种方法。 mplfinance有等价物吗？请帮助 T-T

Answer 1

mplfinace 中的散点图不能单独使用，但可以与烛台结合使用。您的数据被修改为基于月度的数据并用作样本数据。在数据上需要注意的一点是时间序列数据的长度必须相同，否则会出现错误。这个页面很好reference .

import pandas as pd
import numpy as np
import io

data = '''
Date Symbol Action Price
2020-03-01 AAPL Buy 80
2020-04-01 AAPL Sell 130
2020-05-01 AAPL Buy 90
2020-06-01 AAPL Sell 125
2020-07-01 AAPL Buy 125
2020-08-01 AAPL Sell 110
2020-09-01 AAPL Buy 95
2020-10-01 AAPL Sell 125
2020-11-01 AAPL Buy 125
2020-12-01 AAPL Sell 140
2021-01-01 AAPL Buy 115
2021-02-01 AAPL Sell 135
'''

df = pd.read_csv(io.StringIO(data), delim_whitespace=True)

df['Date'] = pd.to_datetime(df['Date'])

buy = df[df['Action'] == 'Buy']
buy2 = df[['Date']].merge(buy,how='outer')
sell = df[df['Action'] == 'Sell']
sell2 = df[['Date']].merge(sell,how='outer')

import mplfinance as mpf
import yfinance as yf

data = yf.download("AAPL", interval='1mo', start="2020-03-01", end="2021-03-01")
data.dropna(how='any', inplace=True)

ap = [mpf.make_addplot(buy2['Price'], type='scatter', marker='^', markersize=200, color='g'),
      mpf.make_addplot(sell2['Price'], type='scatter', marker='v', markersize=200, color='r')
     ]
      
mpf.plot(data, type='candle', ylabel='Candle', addplot=ap, volume=False)