NumPy是一个Python的科学计算库,提供了高性能的多维数组对象,以及一系列的数学函数。NumPy中的flatten()函数是其中的一个函数,可以将多维数组展开成一维数组。
使用方法
flatten()函数的使用方法很简单,只需要将多维数组作为参数传入即可。例如:
import numpy as np a = np.array([[1,2,3],[4,5,6]]) b = a.flatten() print(b)
输出结果为:
[1 2 3 4 5 6]
可以看到,原先的多维数组a被展开成了一维数组b。
参数说明
除了可以传入多维数组作为参数外,flatten()函数还支持三个可选参数:
- order:指定数组的展开顺序,默认为‘C’,表示按行展开;还支持‘F’,表示按列展开。
- dtype:指定输出数组的元素类型,默认为原数组的元素类型。
- copy:指定是否生成一个副本,默认为True,表示生成一个副本;还支持False,表示不生成副本,直接在原数组上进行操作。