Skip to main content
 首页 » 编程设计

python之选择数组中的值

2025年05月04日13zhujiabin

我有两个长度相同的数组:

 x = [2,3,6,100,2,3,5,8,100,100,5] 
 y = [2,3,4,5,5,5,2,1,0,2,4] 

我是这样选择x==100的位置的:

当 x==100 时,y 的值怎么可能? (即 y=5,0,2)? 我这样试过:

 x100=np.where(x==100) 
 y100=y[x100] 

但它没有给我想要的值。我该如何解决这个问题?

请您参考如下方法:

当实际使用numpy 数组 时,您的代码工作正常。你也可以像这样写得更简洁。

>>> import numpy as np 
>>> x = np.array([2,3,6,100,2,3,5,8,100,100,5]) 
>>> y = np.array([2,3,4,5,5,5,2,1,0,2,4]) 
>>> y[x == 100] 
array([5, 0, 2])