时间戳

首页 > 解决方案 > 如何在 Matplotlib/Matplotlib 财务图表上添加补丁？

python - 如何在 Matplotlib/Matplotlib 财务图表上添加补丁？

问题描述

我使用以下内容创建了烛台图：

fig, axlist = mpf.plot(df, type='candle', returnfig=True)

现在我正在尝试添加一个矩形，该矩形从 x 轴的某个点到 x 轴在 y 位置的另一个点。现在，x 轴上的数据是日期，而 y 轴上的数据是价格。

我尝试了以下方法：

ax1 = axlist[0]

new_patch = Rectangle(xy=(0, 9300), width=0.8, height=0.3, angle=0, color='orange')
ax1.add_patch(new_patch)
ax1.autoscale_view()

哪个有效，问题是我不想使用数字来选择 X 的位置。在Rectangle(xy=(0, 9300)使用 0 时，我想使用日期，比如Rectangle(xy=('2020-06-17 10:30', 9300)，但这会给我一个错误：x 必须是一个int。

这是我用来绘制图表的数据框示例：

                                   Date     Open     High      Low    Close      Volume
Date
2020-06-17 19:10:00 2020-06-17 19:10:00  9402.02  9411.03  9400.00  9403.59  215.630925
2020-06-17 19:15:00 2020-06-17 19:15:00  9403.59  9412.54  9403.01  9410.57  108.958008
2020-06-17 19:20:00 2020-06-17 19:20:00  9410.16  9413.66  9409.06  9411.88  107.795579

总结一下：我需要在我的图表上绘制一个补丁，我可以使用一个数字将此补丁放在 x 轴上，但我想使用一个日期。有什么办法吗？

编辑：我尝试了这里建议的解决方案，但没有奏效：

date = datetime.datetime(2020, 6, 19, 10, 30)
testDate = mdates.date2num(date)
print(testDate)

new_patch = Rectangle(xy=(testDate, 9370), width=1, height=0.3, angle=0, color='orange')
ax1.add_patch(new_patch)
ax1.autoscale_view()

因为它给了我一个荒谬的结果：

标签： pythonpython-3.xmatplotlibmplfinance

解决方案

不同的库，不同的实现

mplfinance如果数据框使用 DatetimeIndex，则只能生成绘图；这就是为什么有必要pd.to_datetime事先使用。当您将此图拟合到matplotlib图形中以对其应用补丁时，matplotlib无法将 x 值作为日期处理，因为它们没有被呈现给它。

它所做的是用从 0 开始的整数值列表替换它们。这就是为什么当您将补丁的 x 值分配给testDate737595.4375 时，一旦您自动缩放 x 轴，一切都会被压扁。

解决方案

解决此问题的一种方法是将数据框Date列存储在索引数组中，然后检索匹配日期的索引：

unique_index = pd.Index(df["Date"])
start_x = unique_index.get_loc('2020-06-17 19:10:00')

如果您想在数据框中缺少的日期绘制补丁，您必须找到直接在起点之前的索引并添加到​​ 0 和 1 之间的任何数字。

推荐阅读