python中的.flatten()函数
时间: 2024-06-15 15:04:18 浏览: 195
Python中flatten( )函数及函数用法详解
在Python中,.flatten()函数是用于将多维数组或矩阵转换为一维数组的函数。它可以应用于NumPy数组或Pandas数据框中的数据。
使用.flatten()函数可以将多维数组压缩成一维数组,按照行优先的顺序进行排列。例如,对于一个二维数组,.flatten()函数会将其按照行的顺序展开成一个一维数组。
下面是一个示例代码:
```python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
flatten_arr = arr.flatten()
print(flatten_arr)
```
输出结果为:[1 2 3 4 5 6]
阅读全文