Python中flatten( )函数及函数用法详解

什么是flatten()函数

flatten()函数是一个用于将多维数组“压扁”成一维数组的函数，可以将多维数组转换成一维数组。flatten()函数是Python中的numpy库中的一个函数，因此在使用之前需要先导入numpy库。

函数用法

在numpy中，flatten()函数的使用方式有两种，一种是加括号方式（ndarray.flatten()），一种是调用函数（np.flatten(ndarray)）。

其中，ndarray表示需要压平的多维数组。

使用方式一：

import numpy as np

arr = np.array([[1, 2], [3, 4]])
arr_flatten = arr.flatten()
print(arr_flatten)

# output: [1 2 3 4]

使用方式二：

import numpy as np

arr = np.array([[1, 2], [3, 4]])
arr_flatten = np.flatten(arr)
print(arr_flatten)

# output: [1 2 3 4]

示例说明

下面给出两个示例，帮助读者更好地理解flatten()函数的使用方法。

示例一：将二维数组“压扁”

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
arr_flatten = arr.flatten()
print(arr_flatten)

# output: [1 2 3 4 5 6 7 8 9]

上述代码会将二维数组“压扁”成一个一维数组，并打印出来。

示例二：将三维数组“压扁”

import numpy as np

arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
arr_flatten = arr.flatten()
print(arr_flatten)

# output: [1 2 3 4 5 6 7 8]

上述代码会将三维数组“压扁”成一个一维数组，并打印出来。

经过上述两个示例，我们可以看到flatten()函数的使用非常简单，只需要用数组对象调用这个函数就可以了。同时我们可以看到，flatten()函数是一个对多维数组元素的操作，可以将多维数组“压扁”成一维数组进行处理。