Matplotlib - 使用plt.setp（）的不同颜色

Question

matplotlib.pyplot教程包含以下代码：

  @Override
  public ExitStatus afterStep(StepExecution stepExecution) {
    if(a && b) {
      return new ExitStatus("a");
    } else if(c) {
      return new ExitStatus("b");
    } else if(d) {
      return new ExitStatus("c");
    } else {
      return new ExitStatus("d");
    }
  }

我想知道是否还要为此声明中的不同行指定不同的颜色。

Answer 1

您也可以使用相同的matplotlib.pyplot.setp()方法指定每一行的属性：

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(0, 1.0, 0.01)
y1 = np.sin(2*np.pi*x) # function 1
y2 = np.sin(4*np.pi*x) # function 2
lines = plt.plot(x, y1, x, y2)

l1, l2 = lines # split lines     
plt.setp(l1, linewidth=1, color='r', linestyle='-') # set function 1 linestyle
plt.setp(l2, linewidth=1, color='g', linestyle='-') # set function 2 linestyle

plt.show()

<强>输出：