python - 更改矩阵列的条目

Question

我有以下矩阵:

import numpy as np
A:

matrix([[ 1,  2,  3,  4],
    [ 3,  4, 10,  8]])

问题是我如何输入以下限制条件:如果矩阵A中的任意列数小于或等于(<=)K(3)，则更改最后一个该列的最后一个条目和 5 之间的最小值？所以基本上，我的矩阵应该转换为:

A:

matrix([[ 1,  2,  3,  4],
    [ 3, 4, 5,  8]])

我试过这个功能:

A[-1][np.any(A <= 3, axis=0)] = np.maximum(A[-1], 5)    

但是我有以下错误:

TypeError: NumPy boolean array indexing assignment requires a 0 or 1-dimensional input, input has 2 dimensions

Answer 1

你应该在这里使用 np.minimum。创建掩码和索引，相应地设置值。

B = np.array(A)
m = (B <= 3).any(0)
A[-1, m] = np.minimum(A[-1, m], 5)

A

matrix([[1, 2, 3, 4],
        [3, 4, 5, 8]])