python - Matplotlib - 使用 plt.setp() 的不同颜色

Question

matplotlib.pyplot教程有以下代码:

lines = plt.plot(x1, y1, x2, y2)
# use keyword args
plt.setp(lines, color='r', linewidth=2.0)

我想知道是否可以在此语句中为不同的行指定不同的颜色。

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()

输出: