是否在列表?
使用in
和not 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
。