python 维度拉平
时间: 2023-11-16 11:00:22 浏览: 131
Python3 Tensorlfow:增加或者减小矩阵维度的实现
Python中有多种方法可以将多维数组拉平到一维数组,其中包括使用NumPy和Python自带的函数。下面是两种常用的方法:
1. 使用NumPy的ravel()函数可以将多维数组拉平到一维数组。例如,对于一个二维数组a,可以使用以下代码将其拉平:
```
import numpy as np
a = np.array([[1, 2], [3, 4]])
b = np.ravel(a)
print(b)
```
输出结果为:[1 2 3 4]
2. 使用Python自带的itertools库中的chain()函数和列表推导式可以将多维数组拉平到一维数组。例如,对于一个二维数组a,可以使用以下代码将其拉平:
```
from itertools import chain
a = [[1, 2], [3, 4]]
b = list(chain(*a))
print(b)
```
输出结果为:[1, 2, 3, 4]
阅读全文