Matplotlib：為行指定顏色

Question

我試圖在matplotlib中繪制可變數量的行，其中X，Y數據和顏色存儲在numpy數組中，如下所示。 有沒有辦法將一組顏色傳遞給繪圖函數，所以我不需要額外的步驟為每一行分別分配顏色？ 我是否應該將RGB顏色數組轉換為另一種顏色格式才能使其工作，例如HSV或其他？

import numpy as np
X = np.arange(1990, 1994)
Y = [[  1.50615936e+08   5.88252480e+07   2.60363587e+08]
     [  1.53193798e+08   5.91663430e+07   2.63123995e+08]
     [  1.55704596e+08   5.94899260e+07   2.65840188e+08]
     [  1.58175186e+08   5.97843680e+07   2.68559452e+08]]
colors = [(0.99609375, 0.3984375, 0.3984375) (0.796875, 0.0, 0.99609375)
          (0.59765625, 0.99609375, 0.0)]
#current way
ax.plot(X, Y)
[ax.lines[i].set_color(color) for i, color in enumerate(colors)]
#way I feel it can be done, but doesn't work currently
ax.plot(X, Y, color=colors)
plt.show()

任何幫助是極大的贊賞。

Answer 1

我想你想使用Axes方法set_color_cycle 。 可以想象，它設置了默認情況下分配顏色時循環使用的顏色列表，即沒有為plot調用提供顏色關鍵字時。 這是您的示例的擴展版本：

import matplotlib.pyplot as plt
import numpy as np

X = np.arange(1990, 1994)
Y = [[  1.50615936e+08,   5.88252480e+07,   2.60363587e+08],
     [  1.53193798e+08,   5.91663430e+07,   2.63123995e+08],
     [  1.55704596e+08,   5.94899260e+07,   2.65840188e+08],
     [  1.58175186e+08,   5.97843680e+07,   2.68559452e+08]]
colors = [(0.99609375, 0.3984375, 0.3984375), 
          (0.796875, 0.0, 0.99609375),
          (0.59765625, 0.99609375, 0.0)]

fig = plt.figure()
ax1 = fig.add_subplot(211)
ax1.set_title('old way')
ax1.plot(X, Y)
[ax1.lines[i].set_color(color) for i, color in enumerate(colors)]

ax2 = fig.add_subplot(212)
ax2.set_title('new way')
ax2.set_color_cycle(colors)
ax2.plot(X, Y)

fig.savefig('manycolors.py')
plt.show()

這導致兩個子圖具有相同顏色的線：

Answer 2

自Matplotlib版本1.5以來，仍有一種比@Yann所建議的更新的 “新方式”。 您應該使用set_prop_cycle而不是set_color_cycle使用（depricated）。 在這里你重復他的例子。 我還建議你使用Seaborn ，它有很多預先調制的調色板，你可以選擇顏色的數量。 調色板顏色基於Colorbrewer ，這是一種選擇優質色彩圖的工具。 所以這是我的@Yann代碼版本：

import matplotlib.pyplot as plt
import numpy as np
import seaborn as sns

X = np.arange(1990, 1994)
Y = [[  1.50615936e+08,   5.88252480e+07,   2.60363587e+08],
     [  1.53193798e+08,   5.91663430e+07,   2.63123995e+08],
     [  1.55704596e+08,   5.94899260e+07,   2.65840188e+08],
     [  1.58175186e+08,   5.97843680e+07,   2.68559452e+08]]

colors = sns.color_palette("hls", len(Y[0]))

fig = plt.figure()
ax1 = fig.add_subplot(211)
ax1.set_title('old way')
ax1.plot(X, Y)
[ax1.lines[i].set_color(color) for i, color in enumerate(colors)]

ax2 = fig.add_subplot(212)
ax2.set_title('new way')
ax2.set_prop_cycle('color', colors)
ax2.plot(X, Y)

plt.show()