Python flatten()函数

介绍

flatten()函数是NumPy库中的一个函数，用于将多维数组转换为一维数组。

flatten()函数只能适用于numpy对象，即array或者mat，普通的list列表不适用。

用法

a.flatten(order='C')

其中，order是可选参数，用于指定数组元素在内存中的存储顺序。默认值为'C'，表示按行优先的顺序存储；如果设置为'F'，则按列优先的顺序存储。

实例

import numpy as np

a = np.random.randint(0, 10, size=(2, 3, 4))
b = a.flatten()
c = a.flatten('F')

print('a =', a)
print('b =', b)
print('c =', c)

输出：
a = [[[5 2 3 0]
  [4 1 3 6]
  [4 4 6 5]]

 [[4 3 3 3]
  [3 6 3 8]
  [8 2 7 9]]]
b = [5 2 3 0 4 1 3 6 4 4 6 5 4 3 3 3 3 6 3 8 8 2 7 9]
c = [5 4 4 3 4 8 2 3 1 6 4 2 3 3 3 3 6 7 0 3 6 8 5 9]