将 NumPy 数组转换为 Python 列表
- 2025-01-16 08:38:00
- admin 原创
- 64
问题描述:
如何将NumPy数组转换为 Python 列表?
解决方案 1:
使用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 在评论中指出这一点。)
解决方案 2:
c = np.array([[1,2,3],[4,5,6]])
list(c.flatten())
解决方案 3:
如果 numpy 数组形状是二维,则 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, "
", npa
npa = npa.reshape((3, 3))
print type(npa), npa.shape, "
", npa
a = list(chain.from_iterable(npa))
print type(a), len(a), a`
解决方案 4:
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))
解决方案 5:
另一种选择
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]
也有效。
解决方案 6:
将数组转换为列表的最简单方法是使用 numpy 包:
import numpy as np
#2d array to list
2d_array = np.array([[1,2,3],[8,9,10]])
2d_list = 2d_array.tolist()
要检查数据类型,您可以使用以下命令:
type(object)
相关推荐
热门文章
项目管理软件有哪些?
热门标签
云禅道AD