将numpy.ndarray转化为一维数组
时间: 2023-08-12 17:05:14 浏览: 263
Python中的Numpy入门教程
在 NumPy 中,可以使用 `numpy.ndarray.flatten()` 方法将多维数组(包括二维数组、三维数组等)转化为一维数组。该方法会返回一个新的一维数组,其中包含了原数组中的全部元素,且元素的顺序与原数组中的元素顺序一致。
以下是使用 `flatten()` 方法将二维数组转化为一维数组的示例代码:
``` python
import numpy as np
# 创建一个 2x3 的二维数组
arr2d = np.array([[1, 2, 3], [4, 5, 6]])
# 使用 flatten() 方法将二维数组转化为一维数组
arr1d = arr2d.flatten()
# 打印结果
print("原数组:")
print(arr2d)
print("转化后的一维数组:")
print(arr1d)
```
运行结果如下:
```
原数组:
[[1 2 3]
[4 5 6]]
转化后的一维数组:
[1 2 3 4 5 6]
```
需要注意的是,`flatten()` 方法会返回一个新的一维数组,原数组不会被修改。如果需要修改原数组,可以使用 `numpy.ndarray.ravel()` 方法。但是需要注意的是,`ravel()` 方法返回的是原数组的视图(view),修改视图的值会影响原数组的值,反之亦然。
阅读全文