Jak zmienić shape tablicy

Question

Piszę program, gdzie pobieram sobię obrazki 5x7 pikseli.
Będę chciał je trzymać w takiej tablicy:

inputs = np.ones((number_of_images, 7, 5))

Wrzucam je tablicy tutaj :
 

for i, filename in enumerate(filenames):
  img = Image.open(f"{IMAGES_DIRECTORY}/{filename}")
  inputs[i] = np.array(img)

Jednak dostaję taki error:
 

ValueError: could not broadcast input array from shape (7,5,4) into shape (7,5)

Obrazki składają się tylko z 2 kolorów - czerwony i czarny. Wydaje mi się, że ten 4 wymiar ma coś w wspólnego z kolorami pewnie, ale nie jestem pewny.

Chcę z każdego z nich zrobić po prostu tablicę, gdzie kolor czarny to będzie 1 a kolor biały 0 dla przykładu.

Answer 1

  img = img[:,:,0]