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

Development, Python | Amber | Комментировать запись

Объекты [np.array] в NumPy можно преобразовывать в списки с помощью функции tolist(). Функция tolist() не принимает никаких аргументов. Если массив одномерный, она возвращает список с элементами массива. Для многомерного массива возвращается вложенный список.

Преобразование одномерного массива NumPy в список

Для примера давайте создадим одномерный массив:

import numpy as np

# 1d array to list
arr_1 = np.array([1, 2, 3])

print(f'NumPy Array:\n{arr_1}')

Данный код вернет такой результат: 

NumPy Array:
[1 2 3]

А теперь попробуем применить функцию tolist(): 

import numpy as np

# 1d array to list
arr_1 = np.array([1, 2, 3])

print(f'NumPy Array:\n{arr_1}')

list_1 = arr_1.tolist()

print(f'List: {list_1}')

Новый код вернет: 

List: [1, 2, 3]

Как видите, массив был преобразован в список. 

Преобразование многомерного массива NumPy в список

Давайте теперь построим многомерный массив:

import numpy as np

# 2d array to list
arr_2 = np.array([[1, 2, 3], [4, 5, 6]])

print(f'NumPy Array:\n{arr_2}')

Этот код выведет следующий массив:

NumPy Array:
[[1 2 3]
 [4 5 6]]

А теперь давайте применим функцию tolist():

import numpy as np

# 2d array to list
arr_2 = np.array([[1, 2, 3], [4, 5, 6]])

print(f'NumPy Array:\n{arr_2}')

list_2 = arr_2.tolist()

print(f'List: {list_2}')

Этот код выведет такой результат: 

List: [[1, 2, 3], [4, 5, 6]]

Массив был преобразован из скаляров numpy в скаляры Python.

Заключение

В этой статье мы показали, как использовать функцию tolist() для преобразования объектов np.array в списки. Функцию можно применять как к одномерным, так и к многомерным массивам.

Читайте также:

• Документация API
• Обнаружение и распознавание лиц на фотографиях