堆叠与拆分

NumPy 入门

Izzy Weber

Core Curriculum Manager, DataCamp

切片维度

rgb = np.array([[[255, 0, 0], [255, 255, 0], [255, 255, 255]],
                [[255, 0, 255], [0, 255, 0], [0, 255, 255]],
                [[0, 0, 0], [0, 255, 255], [0, 0, 255]]])
red_array = rgb[:, :, 0]
green_array = rgb[:, :, 1]
blue_array = rgb[:, :, 2]
red_array

array([[255, 255, 255],
       [255,   0,   0],
       [0,   0,   0]])

拆分数组

red_array, green_array, blue_array = np.split(rgb, 3, axis=2)
red_array

array([[[255], [255], [255]],
       [[255], [  0], [  0]],
       [[  0], [  0], [  0]]])

red_array.shape

(3, 3, 1)

末尾维度

red_array_2D = red_array.reshape((3, 3))
red_array_2D

array([[255, 255, 255],
       [255,   0,   0],
       [  0,   0,   0]])

red_array_2D.shape

(3, 3)

数组划分规则

red_array, green_array, blue_array = np.split(rgb, 5, axis=2)

ValueError: array split does not result in an equal division

堆叠数组

将两个二维数组上下堆叠得到三维数组

绘制二维图像数据

red_array, green_array, blue_array = np.split(logo_rgb_array, 3, axis=2)
plt.imshow(red_array)
plt.show()

来自 NumPy 标志的红色通道图

堆叠二维数组

red_array = np.zeros((1001, 1001)).astype(np.int32)
green_array = green_array.reshape((1001, 1001))
blue_array = blue_array.reshape((1001, 1001))

堆叠二维数组

stacked_rgb = np.stack([red_array, green_array, blue_array], axis=2)
plt.imshow(stacked_rgb)
plt.show()

将所有红色 RGB 值替换为 0 的 NumPy 标志

Passons à la pratique !

NumPy 入门