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

298

Як перетворити масив NumPy в список Python (наприклад [[1,2,3],[4,5,6]]) і зробити це досить швидко?

python numpy

426

Використання tolist():

import numpy as np
>>> np.array([[1,2,3],[4,5,6]]).tolist()
[[1, 2, 3], [4, 5, 6]]

Зауважте, що це перетворює значення з будь-якого типу numpy, який вони можуть мати (наприклад, np.int32 або np.float32) у "найближчий сумісний тип Python" (у списку). Якщо ви хочете зберегти нумеровані типи даних, ви можете замість цього зателефонувати список () у своєму масиві, і ви отримаєте список нумерованих скалярів . (Дякую Mr_and_Mrs_D за те, що вказали на це у коментарі.)

— Пітер Хансен
джерело

6

Якщо ваш список float32 tolistперетворить його на floatss - це, мабуть, не бажано. Використання list(myarray)не страждає від цього - чому ми повинні віддавати перевагу tolist?

— Mr_and_Mrs_D

1

@Mr_and_Mrs_D Ви припускаєте, що, незважаючи на те, що плаває у вашому масиві, ви хочете, щоб результат у цьому списку був цілими числами? Це щось зовсім інше питання ... і, звичайно, вам доведеться вказати, що незважаючи на значення float32, всі вони будуть невід'ємними. У будь-якому випадку, якби у вас була така ситуація, я думаю, що це myarray.astype(np.int32).tolist()було б варіантом і явним щодо того, що ви намагаєтеся зробити. (І, нарешті, list(array_of_type_float32)не дає тут цілих чисел, коли я спробував це ... тому я не знаю, що ви просите.)

— Пітер Хансен

1

Я ніколи не згадував цілі числа - спробуйтеfloat32_array = np.array([0.51764709], np.float32); print(float32_array.tolist()); print(list(float32_array))

— Mr_and_Mrs_D

2

Гаразд, тому один перетворюється на [float], а другий все ще використовує numpi скалярний float32 тип, як [np.float32]. Чудово. Добре знати, але я думаю, бажано це чи ні, залежить від кожного конкретного випадку. Я вважаю, що, як правило, коли хтось (наприклад, ОП) запитує перетворення до списку, вони неявно означають список регулярних типів даних Python, в цьому випадку або плаваючі чи цілі числа, а не список числових скалярних типів . Дякуємо, що вказали на це: я відредагував відповідь, щоб включити примітку про це.

— Пітер Хансен

За цими лініями: np.array([[0, 'one'], ['two', 3]]).tolist() -> [['0', 'one'], ['two', '3']]

— keithpjolley

11

Метод numpy .tolist створює вкладені списки, якщо форма масиву numpy 2D.

якщо потрібні плоскі списки, працює метод нижче.

import numpy as np
from itertools import chain

a = [1,2,3,4,5,6,7,8,9]
print type(a), len(a), a
npa = np.asarray(a)
print type(npa), npa.shape, "\n", npa
npa = npa.reshape((3, 3))
print type(npa), npa.shape, "\n", npa
a = list(chain.from_iterable(npa))
print type(a), len(a), a`

— прагнучий Гуру
джерело

5

tolist()працює добре, навіть якщо зіткнувся з вкладеним масивом, скажімо, панди DataFrame;

my_list = [0,1,2,3,4,5,4,3,2,1,0]
my_dt = pd.DataFrame(my_list)
new_list = [i[0] for i in my_dt.values.tolist()]

print(type(my_list),type(my_dt),type(new_list))

— Шивід
джерело

0

someList = [list(map(int, input().split())) for i in range(N)]

— Ганеш Ядхав
джерело

0

c = np.array ([[1,2,3], [4,5,6]])

list(c.flatten())

— Харшал С.Г.
джерело