Python: Wie man NDarray in NDarray-Liste konvertieren

Question

Ich möchte erste Achse numpy.ndarray extrahieren, um eine Liste von numpy.ndarray zu werden.

Zum Beispiel enthält eine arr_A numpy ndarray mit Form (3, 100, 200), die mit der Form einer Liste von numpy ndarray umgewandelt werden (100, 200) und len(arr_A) ist 3.

Was ist einfachste Weg, es zu tun?

Vielen Dank!

Answer 1

Sie können einfach die ndarray an den Konstruktor list passieren:

list_arr_A = list(arr_A) 

Answer 2

list_arr_A = [m for m in arr_A]