リストとNumPy配列を変換

リストをNumPy配列に変換

リストをNumPy配列に変換するには、numpy.array()を使います。

numpyをimportして、

import numpy as np

例えば、下記のリストがあったとします。

listNum = [[1, 2, 3], [2, 2,  5], [3, 4, 7]]  

リストをNumPy配列に変換します。

ndarrayNum = np.array(listNum)

変換後のNumPy配列の中身と型をprint出力してみます。

print(ndarrayNum)
print(type(ndarrayNum))

NumPy配列になっていることが確認できます。

[[1 2 3]
 [2 2 5]
 [3 4 7]]
<class 'numpy.ndarray'>

NumPy配列をリストに変換

NumPy配列をリストに変換するには、
tolist()でできます。

listNum_2 = ndarrayNum.tolist()

変換後のリストの中身と型をprint出力して確認すると良いと思います。

print(listNum)
print(type(listNum))

下記がサンプルコードになります。

$ cat sample.py 
#!/usr/bin/env python3
# coding: UTF-8

import numpy as np

listNum = [[1, 2, 3], [2, 2,  5], [3, 4, 7]]  

print(listNum)
print(type(listNum))

ndarrayNum = np.array(listNum)

print(ndarrayNum)
print(type(ndarrayNum))

listNum_2 = ndarrayNum.tolist()

print(listNum)
print(type(listNum))

$ ./sample.py 
[[1, 2, 3], [2, 2, 5], [3, 4, 7]]
<class 'list'>
[[1 2 3]
 [2 2 5]
 [3 4 7]]
<class 'numpy.ndarray'>
[[1, 2, 3], [2, 2, 5], [3, 4, 7]]
<class 'list'>