numpy ndarray 交换多维数组（矩阵）的行/列

Posted 2021-02-06 siwnhwxh

>> import numpy as np
>> P = np.eye(3)
>> P
array([[ 1.,  0.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  0.,  1.]])

• 1
• 2
• 3
• 4
• 5
• 6

交换第 0 行和第 2 行：

>> P[[0, 2], :] = P[[2, 0], :]
        # P[(0, 2), :] = P[(2, 0), :]
>> P
array([[ 0.,  0.,  1.],
       [ 0.,  1.,  0.],
       [ 1.,  0.,  0.]])

• 1
• 2
• 3
• 4
• 5
• 6

再交换第一列和第三列：

>> P[:, [0, 2]] = P[:, [2, 0]]
>> P
array([[ 1.,  0.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  0.,  1.]])

• 1
• 2
• 3
• 4
• 5
• 6

注意以下是错误的做法：

>> P[0, :], P[2, :] = P[2, :], P[0, :]
>> P
array([[ 0.,  0.,  1.],
       [ 0.,  1.,  0.],
       [ 0.,  0.,  1.]])

        # 不尽写出来啰嗦，而且代表的意义也不是交换

• 1
• 2
• 3
• 4
• 5
• 6
• 7

再分享一下我老师大神的人工智能教程吧。零基础！通俗易懂！风趣幽默！还带黄段子！希望你也加入到我们人工智能的队伍中来！https://blog.csdn.net/jiangjunshow