【Python】【进阶篇】八、Pillo和ndarray数组

八、Pillo和ndarray数组

NumPy 是 Python 科学计算的基础数据包，它被大量的应用于机器学习领域，比如图像识别、自然语言处理、数据挖掘等。
ndarray 是 NumPy 中的数组类型，也称为 ndarray 数组，该数组可以与 Pillow 的 PIL.Image 对象实现相互转化。

8.1 ndarray数组创建图像

下面通过 ndarray 数组构建一个 Image 对象，并显示图像。代码如下：

# 导入相关的包
from PIL import Image
# 使用numpy之前需要提前安装
from numpy import zeros, uint8

# 创建 350*420的图像，3个颜色通道
array = zeros([350, 420, 3], dtype=uint8)
# rgb色彩模式
array[:, :200] = [255, 0, 0]
array[:, 200:] = [255, 255, 0]
img = Image.fromarray(array)
img.show()
img.save("./数组生成图像.png")

8.2 图像转化为ndarray数组

下面通过 ndarray 数组将 Image 对象转化为像素点，并输出像素点。代码如下：

# 导入相关的包
from PIL import Image
# 使用numpy之前需要提前安装
from numpy import array

img = Image.open("hudie2.jpg")
img.show()
# Image图像转换为ndarray数组
img_2 = array(img)
print(img_2)
# ndarray转换为Image图像
arr_img = Image.fromarray(img_2)
# 显示图片
arr_img.show()
# 保存图片
arr_img.save("./arr_img.png")