Как преобразовать массив NumPy в список в Python

Введение

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

Предпосылки

Для выполнения этого урока вам понадобятся:

• Знакомство с установкой Python 3 и использованием pip для установки пакетов. И знакомство с программированием на Python. VS Code для Python.

Это руководство было протестировано с Python 3.9.6 и NumPy 1.23.3.

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

Давайте создадим одномерный массив [1, 2, 3]:

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 в скаляры Python.

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

Давайте создадим многомерный массив [ [1, 2, 3], [4, 5, 6] ]:

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