Numpy ndarray.flatten()函数

在Python中，NumPy是进行科学计算的核心库之一。其中，NumPy数组是NumPy的重要概念。NumPy数组提供了高性能的多维数组对象，并且是大量科学计算的基础。NumPy ndarray.flatten()函数是其中的一个常用函数，它用于将一个多维数组扁平化为一个一维数组。

函数概述

ndarray.flatten(order='C')

参数说明

• order：str类型，可选参数，默认值为'C'，表示数组展开时的顺序。可能的值为'C'、'F'和'A'。'C'表示C语言风格，按行依次展开；'F'表示Fortran风格，按列依次展开；'A'表示根据输入数组的顺序，展开成一维数组。

返回值

返回一个一维数组，该数组包含原多维数组中的所有元素。

使用方法

示例1

import numpy as np

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

print("原多维数组：\n", a)
print("扁平化后的一维数组：\n", b)

原多维数组：
 [[1 2]
 [3 4]]
扁平化后的一维数组：
 [1 2 3 4]

示例2

import numpy as np

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

print("原多维数组：\n", a)
print("按列展开后的一维数组：\n", b)

原多维数组：
 [[1 2]
 [3 4]]
按列展开后的一维数组：
 [1 3 2 4]

示例3

import numpy as np

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

print("原多维数组：\n", a)
print("根据输入数组展开后的一维数组：\n", b)

原多维数组：
 [[1 2]
 [3 4]]
根据输入数组展开后的一维数组：
 [1 2 3 4]

总结

Numpy ndarray.flatten()函数用于将一个多维数组扁平化为一个一维数组。可以根据需要选择展开的顺序，如按C语言风格、Fortran风格或者根据输入数组的顺序展开。常用于一些需要对多维数组进行计算的场景中。