Stapelen en splitsen

Introductie tot NumPy

Izzy Weber

Core Curriculum Manager, DataCamp

Dimensies slicen

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

Arrays splitsen

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)

Achterste dimensies

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)

Regels voor splitsen

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

ValueError: array split does not result in an equal division

Arrays stapelen

Twee 2D-arrays op elkaar stapelen maakt een 3D-array

2D-beelddata plotten

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

Een plot van de rode waarden uit NumPy's logo

2D-arrays stapelen

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

2D-arrays stapelen

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

Een afbeelding van het NumPy-logo met alle rode RGB-waarden vervangen door nullen

Laten we oefenen!

Introductie tot NumPy