Empilhando e dividindo

Introdução ao NumPy

Izzy Weber

Core Curriculum Manager, DataCamp

Fatiando dimensões

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]])

Dividindo arrays

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)

Dimensões finais

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)

Regras de divisão de arrays

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

ValueError: array split does not result in an equal division

Empilhando arrays

Empilhar dois arrays 2D um sobre o outro cria um array 3D

Plotando dados de imagem 2D

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

Um gráfico dos valores de vermelho do logo do NumPy

Empilhando arrays 2D

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

Empilhando arrays 2D

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

Uma imagem do logo do NumPy com todos os valores de vermelho do RGB substituídos por zeros

Vamos praticar!

Introdução ao NumPy