numpy数组如何交换行列？（将BGR变成RGB）

# -*- coding: utf-8 -*-
"""
@File    : numpy_test.py
@Time    : 2020/12/18 9:11
@Author  : Dontla
@Email   : sxana@qq.com
@Software: PyCharm
"""
import numpy as np

arr = np.arange(24).reshape(2, 3, 4)
print(arr)
# [[[ 0  1  2  3]
#   [ 4  5  6  7]
#   [ 8  9 10 11]]
#
#  [[12 13 14 15]
#   [16 17 18 19]
#   [20 21 22 23]]]
print(arr[[1, 0], :, :])  # 交换第一维度的两元素
# [[[12 13 14 15]
#   [16 17 18 19]
#   [20 21 22 23]]
#
#  [[ 0  1  2  3]
#   [ 4  5  6  7]
#   [ 8  9 10 11]]]
print(arr[:, [0, 2, 1], :])  # 交换第二维度的元素
# [[[ 0  1  2  3]
#   [ 8  9 10 11]
#   [ 4  5  6  7]]
#
#  [[12 13 14 15]
#   [20 21 22 23]
#   [16 17 18 19]]]
print(arr[:, :, [0, 2, 1, 3]])  # 交换第三维度元素
# [[[ 0  2  1  3]
#   [ 4  6  5  7]
#   [ 8 10  9 11]]
# 
#  [[12 14 13 15]
#   [16 18 17 19]
#   [20 22 21 23]]]

当然想要删掉某一行、列也是可以的，只需要不指定那行的下标即可

print(arr[:, :, [0, 2, 1]])  # 交换并删除指定行、列
# [[[ 0  2  1]
#   [ 4  6  5]
#   [ 8 10  9]]
# 
#  [[12 14 13]
#   [16 18 17]
#   [20 22 21]]]

参考文章：交换二维numpy数组中的两行和两列