教你flatten()常用的两种用法

介绍

numpy.ndarray.flattern()是用来返回一维数组的函数。

ndarray.flatten(order='C')

用法

常见的order有‘C’,‘F’可以用小写‘c’,‘f’。举个列子说明两种顺序的用法。

import numpy as np

a = np.array([[1,2], [3,4]])

创建一个数组：
$$\left(\begin{array}{c}[1,2]\\ [3,4]\end{array}\right)$$

顺序‘C‘，以行的顺序返回数组。

a.flatten('C')
>array([1, 2, 3, 4])

得到下面一维数组：
$$\left(\begin{array}{c}1\\ 2\\ 3\\ 4\end{array}\right)$$
顺序‘F’，以列的顺序返回数组。

a.flatten('F')
>array([1, 3, 2, 4])

得到下面一维数组：
$$\left(\begin{array}{c}1\\ 3\\ 2\\ 4\end{array}\right)$$

参考资料

numpy.ndarray.flatten