我需要使用 numpy 处理多个条件。
我正在尝试这段似乎有效的代码。
我的问题是:是否有另一种替代方案可以完成同样的工作?
Mur=np.array([200,246,372])*pq.kN*pq.m
Mumax=np.array([1400,600,700])*pq.kN*pq.m
Mu=np.array([100,500,2000])*pq.kN*pq.m
Acreq=np.where(Mu<Mur,0,"zero")
Acreq=np.where(((Mur<Mu)&(Mu<Mumax)),45,Acreq)
Acreq=np.where(Mu>Mumax,60,Acreq)
Print(Acreq)
['0' '45' '60']
最佳答案
从这里开始:
Mur = np.array([200,246,372])*3*5
Mumax = np.array([1400,600,700])*3*5
Mu = np.array([100,500,2000])*3*5
Acreq = np.where(Mu<Mur,0,"zero")
Acreq = np.where((Mur<Mu)&(Mu<Mumax),45,Acreq)
Acreq = np.where(Mu>Mumax,60,Acreq)
print(Acreq)
['0' '45' '60']
试试这个:
conditions = [Mu<Mur, (Mur<Mu)&(Mu<Mumax), Mu>Mumax ]
choices = [ 0, 45, 60 ]
Acreq = np.select(conditions, choices, default='zero')
print(Acreq)
['0' '45' '60']
这也有效:
np.where((Mur<Mu)&(Mu<Mumax),45,np.where(Mu>Mumax,60,np.where(Mu<Mur,0,"zero")))
关于python - Numpy np.where 多重条件,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39307268/