在numpy数组中将负值交换为零

Question

如何编写代码将a的所有负值转换为零?

import numpy as np 
a = np.array([12,12,123,4,-4,0.15,-100])

预期输出:

[12,12,123,4,0,0.15,0]

Answer 1

可以使用numpy中的clip函数

https://numpy.org/doc/stable/reference/generated/numpy.clip.html

a.clip(min = 0)

Answer 2

试试这个:

format_number = lambda n: n if n % 1 else int(n)
a = list(map(lambda n: 0 if n < 0 else format_number(n), a))
print(a)

输出:

[12, 12, 123, 4, 0, 0.15, 0]