3

我正在尝试使用 Matplotlib 绘制曲线族。我直接使用图表绘制数据scatter(),然后使用plot(). 不知道事先会有多少组数据,或者限制等。

我需要能够循环这些线和点的颜色,以便一组数据中的所有内容都匹配。Plot 使用一些内部默认值旋转颜色,并且 scatter 作为所有一种颜色出现。数据集可以靠得很近,所以只是假设从哪些点接近哪条拟合线是不够好的,并且因为我不知道手动制作颜色会有多少条曲线选择是不可扩展的。

此外,因为这些是曲线族(想想晶体管图),我需要能够用曲线显示相关标签。我想做的是将信息写在适合线本身上。

有谁知道其中任何一个的好方法?

4

2 回答 2

13

这试图回答你所有的问题。下面的代码最多循环 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() 

在此处输入图像描述

上面代码中的生成器对于示例来说有点矫枉过正,但它为更复杂的计算提供了结构。如果您只需要几种颜色,则可以使用简单的迭代器

>>> 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'
>>> 

编辑:您有多种选择来获得 n 种不同的颜色。正如我之前指出的,您可以使用其他答案中指示的方法使用生成器。例如,用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 种不同的颜色。

在此处输入图像描述

然而,相似的颜色是连续的,不能提供良好的分辨率/对比度。您可以通过跳过色调值来增加对比度:

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

绘制多条线时的另一个问题是图例......

于 2012-04-21T06:53:49.263 回答
2

我有一个类似的情况,我想为多行提供相同的颜色,同时仍然支持任意数量的行,而无需手动定义它们。这是我想出的生成颜色的函数:

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-20T22:27:37.417 回答