无涯教程-numpy - ndarray.flatten()函数

104 阅读2分钟

在Python中,在某些情况下,我们需要一维数组而不是二维或多维数组。为此,numpy模块提供了一个名为 numpy.ndarray.flatten()的函数,该函数以一维而不是二维或多维数组的形式返回数组的副本。

语法

ndarray.flatten(order=C)

参数:

order:{C,F,A,K}(可选)

如果将order参数设置为C,则意味着该数组以行优先顺序变平。如果设置为" F",则数组将按列优先顺序展平。仅当" a"在内存中是连续的,并且当我们将order参数设置为" A"时,才按列优先顺序对数组进行展平。最后一个顺序是" K",它以与元素在内存中出现的顺序相同的方式展平数组。默认情况下,此参数设置为" C"。

返回值:

y:ndarray

此函数返回源数组的副本,该副本将展平为一维。

范例1:

import numpy as np
a = np.array([[1,4,7], [2,5,8],[3,6,9]])
b=a.flatten()
b

输出:

array([1, 4, 7, 2, 5, 8, 3, 6, 9])

在上面的代码中

  • 我们导入了别名为np的numpy。
  • 我们已经使用 array()函数创建了多维数组a。
  • 我们已经声明了变量b并分配了 flatten()函数的返回值。
  • 最后,我们尝试打印b的值。

在输出中,它显示了一个ndarray,其中包含多维数组的一维元素。

范例2:

import numpy as np
a = np.array([[1,4,7], [2,5,8],[3,6,9]])
b=a.flatten(C)
b

输出:

array([1, 4, 7, 2, 5, 8, 3, 6, 9])

在上面的代码中

  • 我们导入了别名为np的numpy。
  • 我们已经使用 array()函数创建了多维数组a。
  • 我们已经声明了变量b并分配了 flatten()函数的返回值。
  • 我们在函数中使用了C顺序。
  • 最后,我们尝试打印b的值。

在输出中,它显示了一个ndarray,其中包含多维数组的一维元素。

范例3:

import numpy as np
a = np.array([[1,4,7], [2,5,8],[3,6,9]])
b=a.flatten(F)
b

输出:

array([1, 2, 3, 4, 5, 6, 7, 8, 9])

范例4:

import numpy as np
a = np.array([[1,4,7], [2,5,8],[3,6,9]])
b=a.flatten(A)
b

输出:

array([1, 4, 7, 2, 5, 8, 3, 6, 9])

范例5:

import numpy as np
a = np.array([[1,4,7], [2,5,8],[3,6,9]])
b=a.flatten(K)
b

输出:

array([1, 4, 7, 2, 5, 8, 3, 6, 9])

参考链接

www.learnfk.com/numpy/numpy…