作者:鱼儿37度半_795 | 来源:互联网 | 2023-05-21 22:55
假设我有一个10x10矩阵,它包含或仅为0和1,表示为列表列表.我怎么能用matplotlib
这个矩阵来表示红色和黑色方块的网格呢?(红色表示1,黑色表示0).
我已经广泛搜索了,但是我能找到的最接近的是Plot a black-and-white binary map in matplotlib
,但是颜色的浮点数小于1.一旦我得到1的数据,情节就会出错.有人可以帮忙吗?或者指向特定的matplotlib文档,它可以帮助我克服这个问题?
1> Paul H..:
你需要所谓的ListedColorMap
:
import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mpl
# random data
x = np.random.random_integers(0, 1, (10, 10))
fig, ax = plt.subplots()
# define the colors
cmap = mpl.colors.ListedColormap(['r', 'k'])
# create a normalize object the describes the limits of
# each color
bounds = [0., 0.5, 1.]
norm = mpl.colors.BoundaryNorm(bounds, cmap.N)
# plot it
ax.imshow(x, interpolation='none', cmap=cmap, norm=norm)
如果你想要一个能够一次生成色彩映射和规范的函数,还有[`from_levels_and_colors`](http://matplotlib.org/api/colors_api.html#matplotlib.colors.from_levels_and_colors).例如`cmap,norm = from_levels_and_colors([0,0.5,1],['r','k'])`