将NumPy数组转换为Python List结构？

3 回答

• 298

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

  回复于 2024-04-19T03:48:19+08:00
• 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）转换为"nearest compatible Python type"（在列表中） . 如果你想保留numpy数据类型，你可以在你的数组上调用list（），最后你会得到一个numpy scalars的列表 . （感谢Mr_and_Mrs_D在评论中指出这一点 . ）

  回复于 2024-04-19T03:48:19+08:00
• 6

  如果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`
  

  回复于 2024-04-19T03:48:19+08:00