import numpy as np

r = np.array([
    [1, 2, 3, 4],
    [5, 6, 7, 8],
])
g = np.array([
    [11, 22, 33, 44],
    [55, 66, 77, 88],
])
b = np.array([
    [111, 222, 333, 444],
    [555, 666, 777, 888],
])

# složme 3 2D-pole do jednoho 3D-pole (podle třetí osy)
rgb = np.stack((r, g, b), axis=2)

# vzhled a vlastnosti pole
print('rgb:')
print(rgb)
print(rgb.shape, rgb.ndim)

# aplikace výběrové funkce where()
# ~ se třemi parametry provede nahrazení prvků podle vyhodnocení podmínky
print('\n\nrgb % 2 == 1:')
podmínka = (rgb % 2 == 1)
print(podmínka)
print('\n\nwhere(rgb % 2 == 1):')
pole = np.where(rgb % 2 == 1)
print(pole)
print('\n\nwhere(rgb % 2 == 1, 0, rgb):')
pole = np.where(rgb % 2 == 1, 0, rgb)
print(pole)