如何使用matplotlib将RGB颜色值分配给网格网格

前端之家收集整理的这篇文章主要介绍了如何使用matplotlib将RGB颜色值分配给网格网格前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

请考虑这个简化的代码段:

import numpy as np

x = np.arange(3)
y = np.arange(2)

X,Y = np.meshgrid(x,y)

我想将已准备好的颜色值分配给每个网格单元格.
我为每个RGB通道设置了与XY网格相同形状的颜色值作为标准化RGB值 – 因此它是3个numpy数组,每个数组包含0到1个浮点数,表示通道值

我首先尝试使用matplotlib.pyplot.pcolor,因为它似乎是我想要的精确工具,但我无法理解颜色映射是如何完成的.

看起来颜色数组(文档中的C)是在默认(或手动分配)色彩映射上映射的,但是我无法获得此颜色数组中值的逻辑及其作用.
如果值从网格单元格值映射到默认(或指定)色彩映射,那么C颜色数组的目的是什么?
我为上面的例子手工制作了这个C数组(2×1 numpy数组),但不管它是什么值,我都得到相同的颜色,这些颜色似乎只依赖于网格单元格而不是这个C数组中的值.

所以我在这里很困惑并寻求帮助,这不一定需要解释这个pcolor函数,但也许是用matplotlib为网格网格指定颜色值的正确方法是什么

最佳答案
你想要imshow而不是pcolor. (虽然你可以使用pcolor或pcolormesh.)

import numpy as np
import matplotlib.pyplot as plt

# Make some random data to represent your r,g,b bands.
ny,nx = 2,3
r,b = [np.random.random(ny*nx).reshape((ny,nx)) for _ in range(3)]

c = np.dstack([r,b])

plt.imshow(c,interpolation='nearest')
plt.show()

猜你在找的Python相关文章