如何使用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],
[2, 2, 3, 4, 3],
[3, 6, 7, 9, 8],
])
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]
[2 2 3 4 3]
[3 6 7 9 8]]
axis为None,反转行列 [[8 9 7 6 3]
[3 4 3 2 2]
[5 4 3 2 1]]
axis为0,反转行 [[3 6 7 9 8]
[2 2 3 4 3]
[1 2 3 4 5]]
axis为1,反转列 [[5 4 3 2 1]
[3 4 3 2 2]
[8 9 7 6 3]]