python rgba_python-从matplotlib中的RGBA函数创建颜色条

要创建颜色图,您必须指定红色/绿色/蓝色分量在线性比例上的变化方式.看来您已经有了一个函数f,可以为您设置r / g / b组件.最困难的部分是第四个通道,即alpha通道.给定您的f指定的RGB颜色图,我将通过设置Alpha通道进行设置.

import numpy as np

import matplotlib.pyplot as plt

import matplotlib.cm as cm

# some data

a = np.sort(np.random.randn(10, 10))

# use the default 'jet' colour map for showing the difference later

fig = plt.figure()

ax = fig.add_subplot(1, 1, 1)

ax.imshow(a, cmap=cm.get_cmap('jet'))

fig.savefig('map1.png')

# let's use jet and modify the alpha channel

# you would use your own colour map specified by f

my_cmap = cm.get_cmap('jet')

# this is a hack to get at the _lut array, which stores RGBA vals

my_cmap._init()

# use some made-up alphas, you would use the ones specified by f

alphas = np.abs(np.linspace(-1.0, 1.0, my_cmap.N))

# overwrite the alpha channel of the jet colour map

my_cmap._lut[:-3,-1] = alphas

# plot data with our modified colour map

fig = plt.figure()

ax = fig.add_subplot(1,1,1)

ax.imshow(a, cmap=my_cmap)

fig.savefig('map2.png')

这是map1.png：

这是map2.png：

希望这可以帮助.