用numpy查找函数

Question

我有一个numpy数组，我想找到验证某个条件的所有索引。例如，我想绘制Heaviside函数;

import numpy as np
x=np.linspace(-5,5,11)
k_neg=x.find(x<0)
k_pos=x.find(x>=0)
y=np.zeros(len(x))
y(k_neg)=-1
y(k_pos)=1

我没有找到这样的功能（就像在Matlab上一样）。

注意：我的实际问题不是绘制corse的Heavyside;）

Answer 1

Paul Panzer所说;

  

听起来像是在寻找np.where

这解决了我的问题。

Answer 2

我会用numpy在一行中完成：

import numpy as np
x = np.linspace(-5,5,11)
y = ((x>=0)*2)-1