python手动实现图像水平、垂直翻转

投了个实习要求手动实现这个功能，去看了numpy.flip源码简化了一下。。

首先读入图像是shape(row,column,channel)的array数组
例如实例中输入的

img=np.array([[[0, 0, 0],[1,1,1]],[[2,2,2],[3,3,3]]])

实际上是形如下下图的RGB三维数组

水平翻转：

垂直翻转：

这里我们使用np.s_[:]方法

slice(start,end,step)指的是索引从start到end，间隔为step
np.s_[::-1]指的是倒序排列
因为我们普通图像数组是img.ndim维数组，所以设置index = [np.s_[:]]*img.ndim
三个slice分别对应于每一层数组。

import cv2
import numpy as np
def flip(img,axis):
 index = [np.s_[:]]*img.ndim
 index[axis] = np.s_[::-1]
 index = tuple(index)
 return img[index]
if __name__ == "__main__":
 img=cv2.imread("coins.jpg")
 # img=np.array([[[0, 0, 0],[1,1,1]],[[2,2,2],[3,3,3]]])
 cv2.imshow("img",img)
 #axis=0水平翻转
 #axis=1垂直翻转
 cv2.imshow("flip",flip(img,2))
 # print(flip(img,0))
 # print(flip(img,1))
 cv2.waitKey(0)