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

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]