python判断某个元素是否在列表list中以及返回该元素在列表中的索引

文章目录

      • 是否在列表?
      • 返回索引!
      • 总结

是否在列表?

使用innot in关键字即可

number=[1,2,3,4,5]
if 1 in number:
    print("1 in number")
if 0 not in number:
    print("0 not in number")

结果:
在这里插入图片描述

返回索引!

如果一个元素在列表中,可以使用index()来获得索引下标。

number=[1,2,3,4,5,3]
number.index(3)

结果:
在这里插入图片描述
问题:我们发现,明明有2个3,但是却只返回了最前面那个索引。我们可以通过numpy来返回所有。

import numpy as np
number=[1,2,3,4,5,3]
num=np.array(number)
np.argwhere(num==3)

结果:
在这里插入图片描述
注意:转成numpy对列表元素没有要求,但是进行np.argwhere(num==3)判断时,要求num中的所有元素都是数字;同样,如果进行np.argwhere(num==“hello")的时候要求num中的所有元素都是字符串。否则,报错。


总结

对某个元素求index的时候,一定要确保这个元素在列表中,否则报错!

也就是说,如果想查x在列表a中的索引,建议这样使用:

a=[1,2,3,4,5]
x=3
index=-1
#先判断一下x在不在,如果不在,你又去求index,那么将报错。
if x in a:
    index=a.index(x)
print(index)

这样的话,如果元素在其中,获得了第一个index,其大于等于0,否则index=-1

你可能感兴趣的:(python,python,列表)