numpy多维数组转列表之reshape

#!/usr/bin/env python2
# -*- coding: utf-8 -*-


import numpy as np


# example:
data :
array([[-0.35703995,  1.39059152,  0.942042  ,  1.026705  , -0.12740605],
       [-0.40411385,  0.18849482,  1.12767095, -0.82002473,  1.00704771],
       [-0.47190315, -0.85835151, -1.17062813,  0.36865093, -0.83304016],
       [-1.74112618, -1.30695907, -2.6468025 , -1.01027212, -0.84289722]])


# 调整data的形状: eg [4,5] --> [5,4]
data.reshape([5,4]) :
array([[-0.35703995,  1.39059152,  0.942042  ,  1.026705  ],
       [-0.12740605, -0.40411385,  0.18849482,  1.12767095],
       [-0.82002473,  1.00704771, -0.47190315, -0.85835151],
       [-1.17062813,  0.36865093, -0.83304016, -1.74112618],
       [-1.30695907, -2.6468025 , -1.01027212, -0.84289722]])

# 将data转为list,不改变数据的组织结构,使用ndarray.tolist()直接转换即可
data.tolist() :
[[-0.35703994593511124,
  1.3905915232262216,
  0.9420420017892942,
  1.026705002358484,
  -0.12740604886718063],
 [-0.4041138508040703,
  0.1884948205541393,
  1.1276709521986976,
  -0.8200247311598696,
  1.0070477128177504],
 [-0.47190314610448986,
  -0.858351514926757,
  -1.17062812808398,
  0.36865092924302756,
  -0.8330401557493051],
 [-1.741126184186145,
  -1.306959073681527,
  -2.6468024959940846,
  -1.0102721154829577,
  -0.842897223828693]]

# 将data进行重组,得到一维的list:
ret_data = data.reshape(-1).tolist()

# 将一维的list中的元素reshape为多维ndarray
np.array(data.reshape(-1).tolist()).reshape(4,5)
array([[-0.35703995,  1.39059152,  0.942042  ,  1.026705  , -0.12740605],
       [-0.40411385,  0.18849482,  1.12767095, -0.82002473,  1.00704771],
       [-0.47190315, -0.85835151, -1.17062813,  0.36865093, -0.83304016],
       [-1.74112618, -1.30695907, -2.6468025 , -1.01027212, -0.84289722]])