有没有一种简单的方法可以将数组中的所有负值都替换为0?
我对如何使用NumPy数组有一个完整的了解。
例如
a = array([1, 2, 3, -4, 5])
我要回去
[1, 2, 3, 0, 5]
a < 0
给出:
[False, False, False, True, False]
这就是我遇到的问题-如何使用此数组修改原始数组。
Answers:
尝试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])
另一种有趣的方法是使用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放在后面,则使用列表表达式构造的过滤器部分。[(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中完成此操作的方法。创建一个返回所需内容并使用列表推导的函数或map函数。
>>> 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]