np.sign函数怎么用
时间: 2023-08-11 14:07:49 浏览: 144
signal函数的使用
3星 · 编辑精心推荐
np.sign函数是numpy库中的一个函数,其作用是返回输入数组中每个元素的符号。
使用方式如下:
```python
import numpy as np
a = np.array([-2, 0, 3])
b = np.sign(a)
print(b) # [-1 0 1]
```
在上面的例子中,我们首先导入了numpy库,然后创建了一个包含三个元素的一维数组a。接下来,我们使用np.sign函数对数组a中的每个元素进行符号判断,并将结果赋值给一个新的数组b。最后,我们打印出了数组b中的每个元素。
可以看到,np.sign函数返回的是一个新的数组,其中每个元素的值为其对应输入数组元素的符号。如果输入数组元素是正数,则返回1;如果输入数组元素是负数,则返回-1;如果输入数组元素是0,则返回0。
阅读全文