python - python中输出-1或0或1的函数

Question

是否有类似(修改后的)舍入函数，其输出可能为 -1,0,1，具体取决于输入最接近的数字是多少？例如。 0 => 0, -2154 => -1, 10 => 1

目前我使用的是普通的 if else 语句:

if i == 0:
    return 0
elif i > 0: 
    return 1
else:
    return -1

但是有什么办法可以使它成为单行代码吗？例如使用一些修改后的圆形函数。

Answer 1

你可以使用 numpy.sign()

import numpy as np

print(np.sign(1.2))
print(np.sign(-3.4))
print(np.sign(0))

输出:

1.0
-1.0
0

没有任何进口:

def sign(i):
    return (i>0) - (i<0)
    
print(sign(1))
print(sign(-3))
print(sign(0))

输出:

1
-1
0