更改矩阵中的数字

Question

我随机生成一个矩阵。为简单起见，我们假设它采用以下形式np.shape(A) = (2,4)：

import numpy as np
A:

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

那么，我估计如下表达式：

import numpy as np
K = 3
I = 4
C0 = np.sum(np.maximum(A[-1] - K, 0)) / I 

问题是我如何输入以下限制：如果矩阵A中的任何列数小于或等于（

A:

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

Answer 1

这是一种方式。

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

# matrix([[1, 2, 3, 4],
#         [0, 0, 0, 8]])

A[-1][np.any((A > 2) & (A <= 3), axis=0)] = 0

# matrix([[1, 2, 3, 4],
#         [0, 4, 0, 8]])