Python 数组与 RGB 颜色的关系

实验的前提条件

本文会出现一些编程实验，有兴趣实践的读者需要首先确保自己的开发环境安装了 Numpy 和 cv2 包，如果没有的话，可以通过：

pip install numpy
pip install cv2

来进行安装。

---

语境中的Python数组

数组是 Python 语言原生支持的数据结构，直接用方括号声明即可，如：

arr = [1,2,3,4,5]

为了方便数组的创建，Python 支持递推表达式等优雅的写法。在科学计算领域会需要涉及多维数组的运算，普遍使用的是Numpy 包，安装方法请看上一节，公认的用法是

>> import numpy as np

然后用 np 句柄来调用 Numpy 包中的各种函数。我们这篇文章要讨论的数组就是用 Numpy 包创建出来的多维对象。

---

RGB颜色

现代计算机的基本指令和数据都以0和1来表示，出现在显示器上的颜色也是如此。比较基本的做法是用8位二进制数字来表示颜色，数字区间是 [0,255]，这就是常说的“灰度图像“，0表示没有亮度，即黑色，255表示能达到的亮度最大值，即白色。为了表意明确，下面运行一段 Python 脚本来创建三个 $100\times100$ 的灰度图片，然后用系统自带的图片浏览器打开。

# 我们使用cv2.imwrite(filename, img) 方法来将numpy数组写入文件
import cv2
# 跟上文说法一致地导入numpy包
import numpy as np

# 创建一个黑色图片
black = np.zeros((100,100))
cv2.imwrite('black.png', black)

# 创建一个白色图片
white = np.ones((100, 100)) * 255
cv2.imwrite('white.png', white)

# 创建一个灰色图片，灰度值不一定要取120，可以是0到255之间的任意值，读者可以自己改变值来实验一下
gray = np.ones((100, 100)) * 120
cv2.imwrite('gray.png', gray)

运行完成以后当前目录会出现三个文件：white.png, black.png 和 gray.png 打开可以看到：

牛顿的三棱镜实验告诉我们太阳光由七种颜色组成，而计算机显示器表示颜色是由红绿蓝三种颜色组成的，这三种颜色分别称为图片的三个 频道(channel)。可以看到上述生成的图片本质是值在 [0,255] 区间的二维数组，彩色图片的有红绿蓝三个频道，每个频道都是一个二维数组，如果还是规定用8位二进制数来表示一个频道的话，一张彩色图片本质上就是一个值在[0,255]区间的三维数组。我们可以随机生成一个彩色图片，来说明上述概念：

# 这里生成了一个随机数组
img = np.random.randn(100,100,3) * 255
cv2.imwrite('colorful.png', img)

打开图片可以看到：

代码长度和上面生成黑白会的并没有差异，只是数组的形状多了个维度；这里生成的是$100\times100\times3$的数组，正如上文所有，它有三个频道，每个频道都是一个二维数组。np.random.randn(shape) 函数会生成特定形状的随机数组，每个单元的值从-1到1，我们把每个单元的值都扩大了255倍；读者可能会问这里出现负数怎么办，实际上cv2.imwrite自动把负数当正数处理了，我就没有加abs函数。

---

Python数组与RGB颜色的关系 - 逆序

彩色图片的三个频道的顺序跟常识“rgb“恰好相反：

python 数组第三维度索引	对应颜色
0	blue
1	green
2	red

刚才我们生成了随机的彩色图片，现在知道了各个频道与数组第三维度索引之间的对应关系，我们可以来生成一些有趣的新图片了。下面的脚本会生成红色、绿色和蓝色的随机图片，也就是说每个图片只有对应频道有有意义的值，其他频道都取0.

img = np.random.randn(100,100,3) * 255
red = img.copy()
red[:,:,0:2] = 0
cv2.imwrite('red.png', reg)

green = img.copy()
green[:,:,0] = 0
green[:,:,2] = 0
cv2.imwrite('green.png', green)

blue = img.copy()
blue[:,:,1:3] = 0
cv2.imwrite('blue.png', blue)

注意这段代码省略了引入Numpy 和 cv2 包。运行完成以后，当前目录会出现red.png, green.png 和 blue.png 三个图片：

---

练习

读者可以尝试综合以上的内容和自己电脑的分辨率，给自己创建一个新的壁纸，类似这样：

Have fun, cheers!