使用MatPlotLib的顏色和線條寫入

我想繪製使用Matplotlib的曲線族。我使用scatter（）直接繪製數據，然後使用plot（）繪製擬合線（從scipy得到最小二乘）。我不知道事先會有多少組數據，或者這些限制等等。使用MatPlotLib的顏色和線條寫入

我需要能夠循環這些行和點的顏色，以便從一組數據中匹配所有內容。情節使用一些內部默認旋轉顏色和分散出來，所有的一種顏色。數據集可以靠得很近，所以假設從哪個點接近哪個擬合線不夠好，並且因爲我不知道有多少曲線會手動生成一個顏色選擇不可擴展。此外，由於這些曲線族（考慮晶體管圖），我需要能夠用曲線顯示相關標籤。我想要做的是將信息寫在合適的線上。

有誰知道這兩種方法的好方法嗎？

來源

2012-04-20 Elliot

這試圖回答你所有的問題。下面的代碼循環最多7種顏色。如果你需要更多，你應該創造一個更復雜的發電機，如另一個答案所示。

import numpy as np 
from matplotlib import pyplot as plt 

def get_color(): 
    for item in ['r', 'g', 'b', 'c', 'm', 'y', 'k']: 
     yield item 

x = 0.3 * np.array(range(40)) 

color = get_color() 

for group in range(5): 
    # generates a collection of points 
    y = np.exp2(x + 0.5 * group) 
    # fit to a polynomial 
    z = np.polyfit(x, y, 6) 
    p = np.poly1d(z) 

    acolor = next(color) 

    plt.scatter(x, y, color=acolor, marker='o') 
    plt.plot(x, p(x), acolor + '-', label=str(group)) 

plt.legend() 
plt.xlim((0, 15)) 
plt.show()

enter image description here

在上面的代碼發生器是一個位爲例如overkilling的，但它使結構更復雜的計算。如果您只需要幾個顏色，你可以用一個簡單的迭代器

>>> color = iter(list_of_colors) 
>>> acolor = next(color)

，如果你需要循環不休，你可以使用itertools.cycle：

>>> from itertools import cycle 
>>> color = cycle(['r', 'g', 'b', 'c', 'm', 'y', 'k']) 
>>> next(color) 
'r' 
>>>

編輯：您有幾種選擇來獲得不同的顏色。正如我之前指出的，您可以使用其他答案中指示的方法使用發電機。例如，將get_color替換爲不同的發生器：

import colorsys 
import numpy as np 
from matplotlib import pyplot as plt 

def get_color(color): 
    for hue in range(color): 
     hue = 1. * hue/color 
     col = [int(x) for x in colorsys.hsv_to_rgb(hue, 1.0, 230)] 
     yield "#{0:02x}{1:02x}{2:02x}".format(*col) 

x = 0.3 * np.array(range(40)) 

color = get_color(15) 

for group in range(15): 
    # generates a collection of points 
    y = np.exp2(x + 0.5 * group) 
    # fit to a polynomial 
    z = np.polyfit(x, y, 6) 
    p = np.poly1d(z) 

    acolor = next(color) 

    plt.scatter(x, y, color=acolor, marker='o') 
    plt.plot(x, p(x), color=acolor, linestyle='dashed', label=str(group)) 

plt.legend() 
plt.xlim((0, 15)) 
plt.show()

您會得到15種不同的顏色。

enter image description here

類似顏色但連續不給良好的分辨率/對比度。你可以通過跳躍的色調值增加對比度：

for hue in range(0, color*3, 3):

時吸引了許多線是傳說中的另一個問題...

來源

2012-04-21 06:53:49 joaquin

感謝。這解決了眼前的問題，迫使我去了解產量和下一個陳述，而這些陳述我都不知道。但是，如果您需要任意數量的顏色，您會怎麼做？例如，假設我有100個數據集，並且每個數據集都需要完全不同（例如騎自行車不好）。你將如何生成一個顏色迭代器？你可以在沒有明確編碼數據集的情況下做到這一點，所以代碼是可擴展的？ – Elliot 2012-04-24 18:59:42

我有一個類似的情況，我想給多條線相同的顏色，同時仍然支持任意數量的線，而無需手動定義全部線。這是我想出了產生顏色的功能：

import colorsys 

def get_colors(i, total): 
    hue = i*(1.0/total) 
    dark = [int(x) for x in colorsys.hsv_to_rgb(hue, 1.0, 100)] 
    light = [int(x) for x in colorsys.hsv_to_rgb(hue, 1.0, 230)] 
    return "#{0:02x}{1:02x}{2:02x}".format(*dark), "#{0:02x}{1:02x}{2:02x}".format(*light)

正如你所看到的，它產生total顏色與黑暗的最大距離和光版本。

來源

2012-04-20 22:27:37 mensi

使用MatPlotLib的顏色和線條寫入

回答

相關問題