在Matplotlib绘图中突出显示任意点

Question

我是Python和Matplotlib的新手。

我正在尝试在Matplotlib中的一个已有图表中突出显示与特定条件匹配的几个点。

初始图表的代码如下：

pl.plot(t,y)
pl.title('Damped Sine Wave with %.1f Hz frequency' % f)
pl.xlabel('t (s)')
pl.ylabel('y')
pl.grid()
pl.show()

在上面的图中，我想突出显示一些符合 abs(y)>0.5 的特定点。产生这些点的代码如下所示：

markers_on = [x for x in y if abs(x)>0.5]

我尝试使用'markevery'参数，但它会抛出一个错误，提示：

'markevery' is iterable but not a valid form of numpy fancy indexing;

出现错误的代码如下：

pl.plot(t,y,'-gD',markevery = markers_on)
pl.title('Damped Sine Wave with %.1f Hz frequency' % f)
pl.xlabel('t (s)')
pl.ylabel('y')
pl.grid()
pl.show()

- Sra1

Answer 1

“markevery”参数是绘图函数接受不同类型输入的参数。根据输入类型，它们的解释方式也不同。在这个matplotlib示例中可以找到一个很好的可能性列表。

如果您有标记显示的条件，有两个选项。假设“t”和“y”是numpy数组，而且已经导入了“numpy as np”，

1. Either specify a boolean array,

   plt.plot(t,y,'-gD',markevery = np.where(y > 0.5, True, False))
   

或者

2. an array of indices.

   plt.plot(t,y,'-gD',markevery = np.arange(len(t))[y > 0.5])
   

完整示例

import matplotlib.pyplot as plt
import numpy as np; np.random.seed(42)

t = np.linspace(0,3,14)
y = np.random.rand(len(t))

plt.plot(t,y,'-gD',markevery = np.where(y > 0.5, True, False))
# or 
#plt.plot(t,y,'-gD',markevery = np.arange(len(t))[y > 0.5])

plt.xlabel('t (s)')
plt.ylabel('y')

plt.show()

resulting in

- ImportanceOfBeingErnest

---

markevery = np.where(y > 0.5, True, False) 抛出 ValueError: 数组的真值不明确。请使用 a.any() 或 a.all()。 使用 markevery = list(np.where(y > 0.5, True, False)) 可以解决这个问题。 - pfabri

显然，markevery = np.arange(len(t))[y > 0.5] 抛出了完全相同的错误。但是将输出作为 list 传递解决了这个问题。这些错误在 Matplotlib 3.1.3 中抛出。也许自2017年首次发布此问题以来，某个地方发生了重大变化。 - pfabri

@pfabri 感谢通知。这个问题很久以前就出现了。我已经在 https://github.com/matplotlib/matplotlib/pull/17276 中修复了它。 - ImportanceOfBeingErnest

不用谢。我试图编辑您的回答并将我的评论中的修复程序作为替代方案添加进去，但是编辑队列已满。也许您也可以尝试一下...如果您发现我的评论有用...您知道的，我非常需要那些声望点数 :) - pfabri

Answer 2

markevery 使用布尔值来标记每个布尔值为 True 的点。

因此，您可以使用以下代码替换 markers_on = [x for x in y if abs(x)>0.5]： markers_on = [abs(x)>0.5 for x in y] 该代码将返回一个与 y 同样大小的布尔值列表，并且在 |x| > 0.5 的每个点上，您将获得 True 值。

然后，您可以像原始答案一样使用您的代码。

pl.plot(t,y,'-gD',markevery = markers_on)
pl.title('Damped Sine Wave with %.1f Hz frequency' % f)
pl.xlabel('t (s)')
pl.ylabel('y')
pl.grid()
pl.show()

我知道这个问题很老，但在尝试使用最佳答案时，我发现了这个解决方案，因为我不熟悉numpy，它似乎过于复杂。"最初的回答"

- 9 Guy

Answer 3

我遇到了这个问题，因为我试图标记一些超出数据帧边界的点。

例如：

some_df.shape
-> (276, 9)

markers = [1000, 1080, 1120]

some_df.plot(
    x='date',
    y=['speed'],
    figsize=(17, 7), title="Performance",
    legend=True,
    marker='o',
    markersize=10,
    markevery=markers,
)

-> ValueError: markevery=[1000, 1080, 1120] is iterable but not a valid numpy fancy index

请确保您作为标记的值在您想绘制的数据框的范围内。

- Julian Espinel

Answer 4

markevery参数只接受类型为None、整数或布尔数组的索引作为输入。由于我直接传递了值，所以它会抛出错误。

我知道这不是很符合Python的风格，但我使用了下面的代码来得出索引。

marker_indices = []
for x in range(len(y)):
    if abs(y[x]) > 0.5:
        marker_indices.append(x)

- Sra1