python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python matplotlib

详解Python matplotlib中的色彩使用详解

作者:微小冷

matplotlib中提供了一些常见颜色的字符串,并封装成了几个颜色字典,这篇文章主要来和大家讲解一下matplotlib中的色彩使用,需要的可以参考一下

简介

matplotlib中提供了一些常见颜色的字符串,并封装成了几个颜色字典,并放在了matplotlib.colors中,可通过dir(matplotlib.colors)查看,包括

BASE_COLORS为例,其内容为

import matplotlib.colors as mcolors
from pprint import pprint
pprint(mcolors.BASE_COLORS)
# 下面为其返回值
{'b': (0, 0, 1),
 'c': (0, 0.75, 0.75),
 'g': (0, 0.5, 0),
 'k': (0, 0, 0),
 'm': (0.75, 0, 0.75),
 'r': (1, 0, 0),
 'w': (1, 1, 1),
 'y': (0.75, 0.75, 0)}

可见,其基础颜色共8种,分别是

名称RGB名称RGB名称RGB
r(红)(1,0,0)c(青)(0, 0.75, 0.75)k(黑)(0,0,0)
g(绿)(0,0.5,0)m(品红)(0.75, 0, 0.75)w(白)(1,1,1)
b(蓝)(0,0,1)y(黄)(0.75, 0.75, 0)

色彩图绘制函数

下面封装一个函数,来绘制颜色字典。其实很简单,就是将每一种颜色以色块的形式绘制出来,然后在色块后面加上图像的颜色即可。

# 为颜色排序的函数
cSort = lambda c : tuple(mcolors.rgb_to_hsv(mcolors.to_rgb(c)))

而绘制色块,则需要用到Rectangle类,具体实现方法代码如下

import math
from matplotlib.patches import Rectangle
import matplotlib.pyplot as plt
def plot_colortable(colors, ncols=4, sort_colors=True, show=True):
    w,h = 200, 20
    M = 12  # 边缘
    names = list(colors) if not sort_colors else sorted(colors, key=cSort)
    nrows = math.ceil(len(names) / ncols)
    W = w * ncols + 2 * M
    H = h * nrows + 2 * M
    fig, ax = plt.subplots(figsize=(W/72, H/72), dpi=72)
    # 设置图像的边界
    fig.subplots_adjust(M/W, M/H, (W-M)/W, (H-M)/H)
    ax.set_xlim(0, w * ncols)
    ax.set_ylim(h * (nrows-0.5), -h/2.)
    ax.set_axis_off()
    for i, name in enumerate(names):
        row, col = i % nrows, i // nrows
        y = row * h
        st = w * col
        xText = w * col + 55  # 文字的起始位置
        ax.text(xText, y, name, fontsize=14,
                horizontalalignment='left',
                verticalalignment='center')
        rect = Rectangle(xy=(st, y-9), width=48,
                height=18, facecolor=colors[name], edgecolor='0.7')
        ax.add_patch(rect)
    if show: plt.show()
    return fig

其中w和h为每种颜色所占用的宽度和高度,M为其边缘尺寸。输入ncols为列数,通过颜色字典中颜色的个数和ncols计算出所需行数nrow。

色块绘图

最后,分别绘制这四个颜色字典

plot_colortable(mcolors.BASE_COLORS, 3, False)

下图即为基础颜色,总共8种:

plot_colortable(mcolors.CSS4_COLORS, ncols=6)

下图即为CSS颜色,总计148种,光是灰色就多达十来种,只不过gray和grey是同义词,二者的RGB是完全相同的

plot_colortable(mcolors.TABLEAU_COLORS, ncols=5)

下图即为tab颜色,总计10种,分别是灰色、棕色、橘黄色、橄榄色、绿色、青色、蓝色、紫色、粉色以及红色:

XKCD 是由 Randall Munroe 创作的流行网络漫画,他通过20万人调查,得到了将近1000中颜色的名称,并将其进行了RGB映射。matplotlib中的XKCD_COLORS颜色高达949种,采用上述绘图方案得到的图实在太过拥挤,故而将其保存为png

xkcd_fig = plot_colortable(mcolors.XKCD_COLORS)
xkcd_fig.savefig("plt_color_xkcd.png")

结果如下图所示,可见虽然有将近1000中色彩,但并不包含灰色

到此这篇关于详解Python matplotlib中的色彩使用详解的文章就介绍到这了,更多相关Python matplotlib内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文