如何将NumPy数组转换为Python列表?
当前回答
使用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类型”(在列表中)。如果希望保留numpy数据类型,则可以在数组上调用list(),最终将得到一个numpy标量列表。(感谢Mr_and_Mrs_D在评论中指出这一点。)
其他回答
使用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类型”(在列表中)。如果希望保留numpy数据类型,则可以在数组上调用list(),最终将得到一个numpy标量列表。(感谢Mr_and_Mrs_D在评论中指出这一点。)
如果numpy数组形状为2D,则numpy .tolist方法将生成嵌套列表。
如果需要平面列表,下面的方法可以工作。
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`
c = np.array([[1,2,3],[4,5,6]])
list(c.flatten())
tolist()即使遇到嵌套数组也能正常工作,比如pandas 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))
另一个选择
c = np.array([[1,2,3],[4,5,6]])
c.ravel()
#>> array([1, 2, 3, 4, 5, 6])
# or
c.ravel().tolist()
#>> [1, 2, 3, 4, 5, 6]
同样适用。
推荐文章
- 将Pandas或Numpy Nan替换为None以用于MysqlDB
- 使用pandas对同一列进行多个聚合
- 使用Python解析HTML
- django MultiValueDictKeyError错误,我如何处理它
- 如何在for循环期间修改列表条目?
- 我如何在Django中创建一个鼻涕虫?
- 数组与列表的性能
- 没有名为'django.core.urlresolvers'的模块
- 蟒蛇导出环境文件
- Django - makemigrations -未检测到任何更改
- SQLAlchemy:引擎、连接和会话差异
- 在Python Pandas中删除多个列中的所有重复行
- 更改pandas DataFrame中的特定列名
- 将Pandas多索引转换为列
- 熊猫在每组中获得最高的n个记录