78

我有以下问题，我想创建自己的颜色图（red-mix-violet-mix-blue），它映射到 -2 和 +2 之间的值，并想用它来为我的绘图中的点着色。然后，该图应具有右侧的色标。

到目前为止，这就是我创建地图的方式。但我不确定它是否混合了颜色。

cmap = matplotlib.colors.ListedColormap(["red","violet","blue"], name='from_list', N=None)
m = cm.ScalarMappable(norm=norm, cmap=cmap)


这样我将颜色映射到值。

colors = itertools.cycle([m.to_rgba(1.22), ..])


然后我绘制它：

for i in range(0, len(array_dg)):
  plt.plot(array_dg[i], markers.next(),alpha=alpha[i], c=colors.next())


我的问题是：
1. 我无法绘制色标。
2. 我不完全确定我的比例是否正在创建一个连续（平滑）的色阶。

4

4 回答 4

93

这里有一个关于如何创建自定义颜色图的说明性示例。文档字符串对于理解 cdict. 一旦你掌握了它，你可能会使用cdict这样的：

cdict = {'red':   ((0.0, 1.0, 1.0), 
                   (0.1, 1.0, 1.0),  # red 
                   (0.4, 1.0, 1.0),  # violet
                   (1.0, 0.0, 0.0)), # blue

         'green': ((0.0, 0.0, 0.0),
                   (1.0, 0.0, 0.0)),

         'blue':  ((0.0, 0.0, 0.0),
                   (0.1, 0.0, 0.0),  # red
                   (0.4, 1.0, 1.0),  # violet
                   (1.0, 1.0, 0.0))  # blue
          }

尽管该cdict格式为您提供了很大的灵活性，但我发现对于简单的渐变，它的格式相当不直观。这是一个帮助生成简单 LinearSegmentedColormaps 的实用函数：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors as mcolors


def make_colormap(seq):
    """Return a LinearSegmentedColormap
    seq: a sequence of floats and RGB-tuples. The floats should be increasing
    and in the interval (0,1).
    """
    seq = [(None,) * 3, 0.0] + list(seq) + [1.0, (None,) * 3]
    cdict = {'red': [], 'green': [], 'blue': []}
    for i, item in enumerate(seq):
        if isinstance(item, float):
            r1, g1, b1 = seq[i - 1]
            r2, g2, b2 = seq[i + 1]
            cdict['red'].append([item, r1, r2])
            cdict['green'].append([item, g1, g2])
            cdict['blue'].append([item, b1, b2])
    return mcolors.LinearSegmentedColormap('CustomMap', cdict)


c = mcolors.ColorConverter().to_rgb
rvb = make_colormap(
    [c('red'), c('violet'), 0.33, c('violet'), c('blue'), 0.66, c('blue')])
N = 1000
array_dg = np.random.uniform(0, 10, size=(N, 2))
colors = np.random.uniform(-2, 2, size=(N,))
plt.scatter(array_dg[:, 0], array_dg[:, 1], c=colors, cmap=rvb)
plt.colorbar()
plt.show()

在此处输入图像描述

顺便说一句，for-loop

for i in range(0, len(array_dg)):
  plt.plot(array_dg[i], markers.next(),alpha=alpha[i], c=colors.next())

每次调用plt.plot. 这将适用于少数点，但对于许多点会变得非常慢。plt.plot只能绘制一种颜色，但plt.scatter可以为每个点分配不同的颜色。因此，plt.scatter是要走的路。

于 2013-05-30T12:27:49.917 回答
92

由于其他答案中使用的方法对于如此简单的任务似乎相当复杂，因此这里有一个新答案：

代替ListedColormap产生离散颜色图的 a ，您可以使用 a LinearSegmentedColormap。这可以很容易地使用该from_list方法从列表中创建。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors

x,y,c = zip(*np.random.rand(30,3)*4-2)

norm=plt.Normalize(-2,2)
cmap = matplotlib.colors.LinearSegmentedColormap.from_list("", ["red","violet","blue"])

plt.scatter(x,y,c=c, cmap=cmap, norm=norm)
plt.colorbar()
plt.show()

在此处输入图像描述

更一般地说，如果您有一个值列表（例如[-2., -1, 2]）和相应的颜色（例如["red","violet","blue"]），这样n第 th 值应该对应于第nth 颜色，您可以规范化这些值并将它们作为元组提供给from_list方法。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors

x,y,c = zip(*np.random.rand(30,3)*4-2)

cvals  = [-2., -1, 2]
colors = ["red","violet","blue"]

norm=plt.Normalize(min(cvals),max(cvals))
tuples = list(zip(map(norm,cvals), colors))
cmap = matplotlib.colors.LinearSegmentedColormap.from_list("", tuples)

plt.scatter(x,y,c=c, cmap=cmap, norm=norm)
plt.colorbar()
plt.show()

在此处输入图像描述

于 2017-10-16T20:16:58.253 回答
14

如果您想自动创建通常用于表面图的自定义发散颜色图，此模块与 @unutbu 方法相结合对我来说效果很好。

def diverge_map(high=(0.565, 0.392, 0.173), low=(0.094, 0.310, 0.635)):
    '''
    low and high are colors that will be used for the two
    ends of the spectrum. they can be either color strings
    or rgb color tuples
    '''
    c = mcolors.ColorConverter().to_rgb
    if isinstance(low, basestring): low = c(low)
    if isinstance(high, basestring): high = c(high)
    return make_colormap([low, c('white'), 0.5, c('white'), high])

高低值可以是字符串颜色名称或 rgb 元组。这是使用曲面图演示的结果： 在此处输入图像描述

于 2015-06-07T01:43:52.750 回答
2

这似乎对我有用。

def make_Ramp( ramp_colors ): 
    from colour import Color
    from matplotlib.colors import LinearSegmentedColormap

    color_ramp = LinearSegmentedColormap.from_list( 'my_list', [ Color( c1 ).rgb for c1 in ramp_colors ] )
    plt.figure( figsize = (15,3))
    plt.imshow( [list(np.arange(0, len( ramp_colors ) , 0.1)) ] , interpolation='nearest', origin='lower', cmap= color_ramp )
    plt.xticks([])
    plt.yticks([])
    return color_ramp

custom_ramp = make_Ramp( ['#754a28','#893584','#68ad45','#0080a5' ] )

自定义色带

于 2020-06-02T22:35:01.047 回答