Python 中的 numpy.ndarray.flatten()

2024 年 8 月 29 日 | 阅读 2 分钟

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

语法

参数

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

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

返回值

y: ndarray

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

示例 1

输出

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

在上面的代码中

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

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

示例 2

输出

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

在上面的代码中

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

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

示例 3

输出

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

示例 4

输出

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

示例 5

输出

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

下一主题Numpy.meshgrid()