Pyplot-如果数据小于零，更改线的颜色？

Question

我试图找出是否有任何内置的pyplot会改变我的行的颜色，取决于数据是负还是正。例如，如果为负，我希望线条为红色，如果为正，我希望线条为不同的颜色，例如黑色。

库中有什么可以让我执行此操作的吗？我想到的一件事是将数据分为正数和负数两组，并分别绘制它们，但我想知道是否有更好的方法。

Answer 1

我只是制作两个数据集并设置正确的蒙版。通过使用这种方法，我不会在不同的积极方面之间划清界限。

import matplotlib.pyplot as plt
import numpy as np

signal = 1.2*np.sin(np.linspace(0, 30, 2000))
pos_signal = signal.copy()
neg_signal = signal.copy()

pos_signal[pos_signal <= 0] = np.nan
neg_signal[neg_signal > 0] = np.nan

#plotting
plt.style.use('fivethirtyeight')
plt.plot(pos_signal, color='r')
plt.plot(neg_signal, color='b')
plt.savefig('pos_neg.png', dpi=200)
plt.show()