配列内のすべての負の値を0に置き換える簡単な方法はありますか?
NumPy配列を使用してそれを行う方法について完全なブロックがあります。
例えば
a = array([1, 2, 3, -4, 5])
戻る必要があります
[1, 2, 3, 0, 5]
a < 0
与える:
[False, False, False, True, False]
これは私が立ち往生しているところです-この配列を使用して元の配列を変更する方法。
回答:
試してみてくださいnumpy.clip
:
>>> import numpy
>>> a = numpy.arange(-10, 10)
>>> a
array([-10, -9, -8, -7, -6, -5, -4, -3, -2, -1, 0, 1, 2,
3, 4, 5, 6, 7, 8, 9])
>>> a.clip(0, 10)
array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
で下半分のみをクリップできますclip(0)
。
>>> a = numpy.array([1, 2, 3, -4, 5])
>>> a.clip(0)
array([1, 2, 3, 0, 5])
で上半分のみをクリップできますclip(max=n)
。(これはNaN
、最初のパラメーターに渡してout
型を強制するために使用するという以前の提案よりもはるかに優れています。):
>>> a.clip(max=2)
array([ 1, 2, 2, -4, 2])
もう1つの興味深いアプローチは、以下を使用することwhere
です。
>>> numpy.where(a <= 2, a, 2)
array([ 1, 2, 2, -4, 2])
最後に、aixの答えを考えてみましょう。clip
自己文書化であるため、単純な操作が好きですが、より複雑な操作には彼の答えが望ましいです。
numpy
?あなたは、クリップしてみてくださいましたメソッドのをa
?組み込み関数でnumpy.clip
も同じエラーが発生しますが、メソッドでは発生しません。
numpyを使用しない別のミニマリストPythonソリューション:
[0 if i < 0 else i for i in a]
追加の関数を定義する必要はありません。
a = [1, 2, 3, -4, -5.23, 6]
[0 if i < 0 else i for i in a]
収量:
[1, 2, 3, 0, 0, 6]
for
適用の前に置くと、リストのすべての要素に適用され、後に置くと、条件が満たされた場合にのみ、結果のリストに入れられます。
i < 0 ? 0 : i
リスト内包表記内で(Cの)三項演算子を使用しているだけです。わかりやすくするために角かっこを入れます[(0 if i < 0 else i) for i in a]
。if afterを置くことは、リスト式コンストラクトのフィルター部分を使用しています。[(i) for i in a if i < 0]
ゼロ未満のアイテムのリストのみを返します。
そしてさらに別の可能性:
In [2]: a = array([1, 2, 3, -4, 5])
In [3]: where(a<0, 0, a)
Out[3]: array([1, 2, 3, 0, 5])
NumPyを使用せずにPythonでそれを行う方法は次のとおりです。必要なものを返す関数を作成し、リスト内包表記またはマップ関数を使用します。
>>> a = [1, 2, 3, -4, 5]
>>> def zero_if_negative(x):
... if x < 0:
... return 0
... return x
...
>>> [zero_if_negative(x) for x in a]
[1, 2, 3, 0, 5]
>>> map(zero_if_negative, a)
[1, 2, 3, 0, 5]