python - 如果信号值大于信号平均值则画一条线

Question

当我的信号值大于该信号的平均值时，我想在图表上画一条线。我的信号是肌肉拉动和放松的简单串联。

muscle_on2 = np.random.uniform(-1, 1, 1000) + offs
muscle_off = np.random.uniform(-0.05, 0.05, 500) + offs
emg = np.concatenate([muscle_off, muscle_on1, muscle_off, muscle_on2, muscle_off])

为了做到这一点，我尝试了这样的事情:

pl.plot(x_axis)
for i in range(0, len(emg), 1):
    if emg[i]>RM:
        pl.plot([i, 1.0], [i, 1.0], 'r-', lw=5)

其中 RM 是:

RM = np.mean(emg)

但是，它不是生成水平线，而是生成垂直线:

有人可以帮我解决这个问题吗？

我忘记澄清这一点，在这种情况下offs(偏移量已经被过滤掉)这部分代码的实现是为了显示我的信号是如何创建的。当我打印出照片上的平均值时，它在 0.26 左右波动。我希望我的线在信号值大于平均值时出现，并在信号值小于平均值时结束。

Answer 1

您应该使用 matplotlib.pyplot.axhline ,

import matplotlib.pyplot as plt
import numpy as np

muscle_on2 = np.random.uniform(0, 1, 1000)
muscle_off = np.random.uniform(0, 0.05, 500)
emg = np.concatenate([muscle_off, muscle_on2, muscle_off])
RM = np.mean(emg)
plt.plot(emg)
plt.axhline(RM, color='r')
plt.show()

这会给你

或matplotlib.pyplot.hlines如果你不想一路走完剧情，

import matplotlib.pyplot as plt
import numpy as np

muscle_on2 = np.random.uniform(0, 1, 1000)
muscle_off = np.random.uniform(0, 0.05, 500)
emg = np.concatenate([muscle_off, muscle_on2, muscle_off])
e = x[emg > RM]
plt.plot(x, emg)
plt.hlines(RM, e[0], e[-1], color='r', lw=2, zorder=3)
plt.show()

<小时/>

注意   -   matplotlib.pyplot 的导入约定是

import matplotlib.pyplot as plt

以及 deprecated 的约定pylab 是

import matplotlib.pylab as pl

如果您正在使用 pylab 那么您应该切换到 pyplot 作为 official usage guide表示。