如果信号的值大于信号的平均值，则画一条线

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)

然而，它不是产生水平线，而是产生垂直线：

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

我忘了澄清这一点，在这种情况下，偏移量（偏移量已被过滤掉）这部分代码是为了显示我的信号是如何创建的。当我打印出照片上的平均值时，它在 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()

---

注意   -   import 对 matplotlib.pyplot 的约定是

import matplotlib.pyplot as plt

deprecated pylab 的约定是

import matplotlib.pylab as pl

如果您使用的是pylab，那么您应该按照official usage guide 的指示切换到pyplot。