numpy反转数组

numpy反转数组

1、使用np.flip()函数

使用flip(m, axis=None)函数对数组进行反转：

m:输入数组
axis:为None则行列都反转
axis:为0则反转行
axis:为1则反转列

2、代码

import numpy as np

# 创建一维数组
arr = np.array([
    [1, 2, 3, 4, 5],
    [4, 3, 2, 2, 1],
    [6, 7, 8, 9, 0],
])
print('原数组', arr)
# 使用 flip 函数反转数组
reversed_arr = np.flip(arr)
print('axis为None，反转行列', reversed_arr)

reversed_arr = np.flip(arr, axis=0)
print('axis为0，反转行', reversed_arr)

reversed_arr = np.flip(arr, axis=1)
print('axis为1，反转列', reversed_arr)

输出：

原数组 [[1 2 3 4 5]
 [4 3 2 2 1]
 [6 7 8 9 0]]
axis为None，反转行列 [[0 9 8 7 6]
 [1 2 2 3 4]
 [5 4 3 2 1]]
axis为0，反转行 [[6 7 8 9 0]
 [4 3 2 2 1]
 [1 2 3 4 5]]
axis为1，反转列 [[5 4 3 2 1]
 [1 2 2 3 4]
 [0 9 8 7 6]]