Python中numpy的翻转数组
时间: 2024-04-20 15:05:07 浏览: 85
python 实现旋转数列
翻转数组(即将数组中的元素按照某个方向进行翻转)可以使用numpy的flip函数来实现。flip函数可以在指定的轴上翻转数组的元素。
下面是一个示例代码,将一个二维数组沿着水平方向翻转:
```python
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print("原数组:")
print(a)
# 水平方向翻转
b = np.flip(a, axis=1)
print("水平翻转后的数组:")
print(b)
```
输出结果如下:
```
原数组:
[[1 2 3]
[4 5 6]
[7 8 9]]
水平翻转后的数组:
[[3 2 1]
[6 5 4]
[9 8 7]]
```
同样地,也可以在垂直方向翻转数组,只需要将axis参数设置为0即可:
```python
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print("原数组:")
print(a)
# 垂直方向翻转
b = np.flip(a, axis=0)
print("垂直翻转后的数组:")
print(b)
```
输出结果如下:
```
原数组:
[[1 2 3]
[4 5 6]
[7 8 9]]
垂直翻转后的数组:
[[7 8 9]
[4 5 6]
[1 2 3]]
```
阅读全文