python - 関数呼び出し中に更新される間、行列要素は最大値を超えてはなりません

Question

以下のfnを使用しています：

def TABs4_tmp(x,ee,M,nrm1):  
    M += dot(sign(ee),sign(x.T))/nrm1 

ex:
      nrm1 is scalar= 2^10  
      ee is scalar= 1  
      x is an array of 100 elements

プログラムでこの関数を何度も呼び出しており、M 配列 (サイズ 100) の各要素の値が毎回更新されています。現在、M の要素は任意の値を取ることができますが、要素の値を最大値によって制限する必要があります。 2を参照してください。

M の要素のうち、値が 2 未満の要素は、この関数を呼び出している間に新しい値に更新できますが、最大値 2 に達した要素は更新しないでください。

私は Python を初めて使用します。解決策を提案するのを手伝ってください。

Answer 1

Numpy にはまさにこのための機能があります。

これらはMaskedArrayと呼ばれます。

import numpy.ma as ma

M = ma.MaskedAray(M)

def TABs4_tmp(x,ee,M,nrm1):
    M = ma.masked_greater(2,M)
    M += dot(sign(ee),sign(x.T))/nrm1 

これは、マスクされていない部分にのみ追加されます。