如何在另一条线图上绘制特定点?

时间:2021-05-05 16:46:55

标签: python pandas dictionary matplotlib plot

我是一名交易员,我想跟踪标的证券价格变动中的所有头寸入场点。例如,这是迄今为止我的 BTC 头寸代码。我导入我的库,加载我的数据并绘制最近几个月收盘价数据的范围。一切都是我想要的,但我不知道如何绘制我的位置条目。

这是代码和输出:

# Libraries
import pandas as pd
import matplotlib.pyplot as plt
import matplotlib.ticker as mtick
import yfinance as yf


# Data load
BTC = yf.download(tickers='BTC-USD', start='2020-10-01', end='2021-05-03')
BTC_close = BTC['Close']
print(BTC.head())
BTC_positions = yf.download(tickers='BTC-USD', start='2021-02-01', end='2021-05-03')
MyBTC = BTC_positions['Close']
df_RAW = pd.DataFrame(BTC_positions, columns=['Date','Close'])



# Plot
fig = plt.figure()
f, ax = plt.subplots(1,1)
ax.tick_params(length=3, direction='in', labelsize=10)
fmt = '${x:,.0f}'
tick = mtick.StrMethodFormatter(fmt)
ax.yaxis.set_major_formatter(tick)
ax.xaxis.set_tick_params(direction='in')
MyBTC.plot(figsize=(12, 7))
plt.xticks(rotation=0)
plt.title("Bitcoin Position Entries", fontsize=20)
plt.xlabel('Day',fontsize=13) # or fontsize=10
plt.ylabel('Price $$$',fontsize=13) # or fontsize=10
plt.show()

输出图: 我想在其上绘制条目的原始数据图 Raw data plot

我尝试合并 2 个不同的词典:

字典 1 的键:值对,输入日期:花费金额

Dict1 = {
    '2021-02-17': 250,
    '2021-02-24': 250,
    '2021-02-28': 50,
    '2021-03-08': 500,
    '2021-04-10': 500,
    '2021-04-14': 120,
    '2021-04-17': 400,
    '2021-04-21': 120,
    '2021-04-22': 875,
    '2021-04-23': 300,
    '2021-04-29': 125
}

具有键值对的字典 2 的输入日期:基础证券的价格

Dict2 = {
    'Date': ['2021-02-17','2021-02-24','2021-02-28','2021-03-08','2021-04-10','2021-04-14','2021-04-17','2021-04-21','2021-04-22','2021-04-23','2021-04-29'],
    'Price': [51280, 49426, 44548, 50840, 60251, 62479, 53637, 55946, 51376, 48080, 53955]
}

我面临的问题是我在 2 月 17 日购买了价值 250 美元的商品,但我希望该点以适当的入场价格显示,即 51,280 美元,因此 y 轴超出比例。 我想我可以绘制点,并根据购买的数量改变它们的大小或颜色渐变(较小的购买 = 小/亮点,较大的购买 = 大/暗点)

我尝试创建一个 DataFrame 是因为我认为这会更容易,但我只是让自己更加困惑。

我已经走到了这一步,完全不知道如何去做。我以为我使用 Dict1 和 Dict2 是在正确的轨道上,但现在我感到更加困惑。任何有关如何解决此问题的帮助将不胜感激。

1 个答案:

答案 0 :(得分:0)

迭代 Dict1.items() 以从 date 数据框中获取 CloseMyBTC 价格(您也可以使用 annotate 添加您想要的价格从Dict1购买)。然后,应用 scatter 函数在现有线图的顶部绘制特定点。此函数将标记放置在 x 轴中由 date 指定的位置和 y 轴中的收盘价(closep 来自 MyBTC),并且像往常一样对于某些 matplotlib ,您可以设置标记形状、大小 (linewidths) 和颜色。

...
...
plt.xlabel('Day',fontsize=13) # or fontsize=10
plt.ylabel('Price $$$',fontsize=13) # or fontsize=10

for date,buy_price  in Dict1.items():
    closep = MyBTC[date]
    plt.annotate(str(buy_price), (date, closep), xytext=(-3, 6),
            textcoords='offset points',  weight='bold', fontsize=12)
    plt.scatter(date, closep, marker='v', linewidths=1, color='r')

plt.show()

plot_with_markers