web-dev-qa-db-ja.com

NumPy配列をPythonのリスト構造に変換する

NumPy 配列をPythonリスト(例えば[[1,2,3],[4,5,6]])に変換するにはどうすればよいですか。

229
Alex Brooks

tolist() を使用してください。

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

これはそれらが持っているかもしれないどんなでこぼこしたタイプ(例えばnp.int32またはnp.float32)から(リスト内の)最も近い互換性のあるPythonタイプへ値を変換することに注意してください。あなたが派手なデータ型を保存したい場合は、代わりにあなたの配列でlist()を呼び出すことができます、そしてあなたは 派手なスカラのリストになるでしょう (コメントでそれを指摘してくれたMr_and_Mrs_Dに感謝します。)

338
Peter Hansen

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`
8
aspiringGuru

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))
2
Hamidreza

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

0
Ganesh Jadhav