替换numpy数组中的负值


90

有没有一种简单的方法可以将数组中的所有负值都替换为0?

我对如何使用NumPy数组有一个完整的了解。

例如

a = array([1, 2, 3, -4, 5])

我要回去

[1, 2, 3, 0, 5]

a < 0 给出:

[False, False, False, True, False]

这就是我遇到的问题-如何使用此数组修改原始数组。

Answers:


137

你在那儿 尝试:

In [4]: a[a < 0] = 0

In [5]: a
Out[5]: array([1, 2, 3, 0, 5])

89

尝试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简单的操作,因为它是自记录的,但是对于更复杂的操作,他的答案是更可取的。


1
a.clip(0)就足够了,因为OP只是想替换负值。a.clip(0,10)将上述排除任何10.
兔子

1
@Hiett-我刚刚尝试过,剪辑需要一个。首先假定最小。
Usagi '04

必须是numpy的版本问题-这是我的输出:(Pdb)np.clip(w,0)*** TypeError:clip()至少接受3个参数(给定2个)-而:(Pdb)np.clip( w,0,1e6)array([[0.,0.605]])
bph

1
@Hiett,什么版本的numpy?你尝试剪辑方法a?内置函数numpy.clip给了我同样的错误,但是方法却没有。
senderle'4

是的,如果您这样称呼它似乎可以正常工作,例如p w.clip(0)array([[0.,0.605]])-奇怪吗?
bph 2012年

10

另一个不使用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]

1
很好-我想知道将if语句放入列表理解中的语法是什么-我在for循环后粘贴它,然后又返回了两个值,例如,您的示例为[0,0],这是错误的列表
bph 2012年

当我最初了解列表理解并且尝试不同的方法来测试我的理解时,我做了同样的事情-将它放在for循环之后对我来说似乎更直观。但是,现在这样做了:)将它放在将它for应用于列表的每个元素之前,放在后面,意味着只有在满足条件的情况下,它才会进入结果列表。
Levon 2012年

2
@Hiett只是i < 0 ? 0 : i在列表推导中使用三元运算符(在C中)。放入方括号使其更清晰[(0 if i < 0 else i) for i in a]。如果将if放在后面,则使用列表表达式构造的过滤器部分。[(i) for i in a if i < 0]只会返回小于零的项目列表。
Paul S

2
Numpy功能强大,因为它通过编译的C代码执行大量计算,因此速度更快。与其他方法相比,我发现速度因子差异几乎是10倍(这是较慢的)。因此,尽管直观且易于阅读,但这绝对不是计算密集型的。
rspencer


2

这是在没有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]

1
已经走了这条路线,但认为必须使用numpy来实现一种更简单,使用matlab更少的python方法(因为我无论如何都使用数组而不是列表)。剪辑是完美的
bph 2012年
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.