Метод tolist класса ndarray
Метод tolist преобразует массив NumPy в список Python. Метод не принимает параметров. Возвращает вложенный список, который повторяет структуру размерностей исходного массива. Для многомерных массивов возвращается вложенный список списков. Элементы преобразуются в соответствующие типы Python.
Синтаксис
arr.tolist()
Пример
Преобразуем одномерный массив в список:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
res = arr.tolist()
print(res)
print(type(res))
Результат выполнения кода:
[1, 2, 3, 4, 5]
<class 'list'>
Пример
Преобразуем двумерный массив во вложенный список:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
res = arr.tolist()
print(res)
print(type(res))
print(type(res[0]))
Результат выполнения кода:
[[1, 2, 3], [4, 5, 6]]
<class 'list'>
<class 'list'>
Пример
Преобразуем трехмерный массив:
import numpy as np
arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
res = arr.tolist()
print(res)
Результат выполнения кода:
[[[1, 2], [3, 4]], [[5, 6], [7, 8]]]
Пример
Сравнение с преобразованием через list():
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
list_res = list(arr)
tolist_res = arr.tolist()
print("list():", list_res)
print("tolist():", tolist_res)
Результат выполнения кода:
list(): [array([1, 2, 3]), array([4, 5, 6])]
tolist(): [[1, 2, 3], [4, 5, 6]]