Python numpy.flatten()

概述

numpy.flatten()函数用于将一个多维数组转换为一维数组。它返回一个新的数组，该数组包含原始数组中的所有元素，并按照行优先顺序排列。

语法：

numpy.flatten(order='C')

参数：

• order（可选）： 'C'表示按行顺序返回数组，'F'表示按列顺序返回数组。

返回值：

一个一维数组。

示例

示例1

import numpy as np

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

输出：

[[1 2]
 [3 4]]
[1 2 3 4]

示例2

import numpy as np

a = np.array([[1, 2], [3, 4]])
print(a)
b = a.flatten(order='F')
print(b)

输出：

[[1 2]
 [3 4]]
[1 3 2 4]

总结

numpy.flatten()函数可以方便地将多维数组转换为一维数组。如果需要改变返回数组的顺序，可以使用order参数。