Преобразование массива NumPy в структуру списка Python

Question

Как мне преобразовать массив numpy в список Python?например,

y = array([[0],[0],[1],[0],[0]])

y_list = [0,0,1,0,0]

Я попытался y.tolist() и вывод: y=[[0],[0],[1],[0],[0]]

Answer 1

Используйте flatten(), чтобы сгладить массив - от array([[1,2], [3,4]]) до array([1,2,3,4]), а затем преобразовать полученный array в list, используя функцию list().

y_list = list(y.flatten('F'))

Answer 2

Вы можете использовать ravel и tolist:

y.ravel().tolist()