如何遍历前 n 个元素

Question

我正在尝试遍历不同长度的不同数组的前 n 个元素，以检查该数字是否存在

a=[0,1,2,4,5]
count=0
for i in range(3):
  if a[i]==4:
    count+=1
if count>=1:
  print(count)
else:
  print('not in array')

我也试过这个

for in range a[:3]:

问题是如果数组长度小于 n，在本例中为 3。

Answer 1

使用

列表切片count怎么样？

a[:n].count(4)

这会自动处理结束条件，因此不存在数组长度小于 n 的问题。

---

如果您只需要成员资格检查，请在集合上使用成员资格运算符in(如其他应答者所示(：

4 in set(a[:n])

Answer 2

要检查元素是否在数组中，您可以使用集合：

>>> a = [0,1,2,4,5]
>>> 4 in set(a[:3])
False
>>> 3 in set(a[:3])
True

Answer 3

尝试在 for 循环中使用 len() 函数：

a=[0,1,2,4,5]
count=0
for i in range(len(a)):
  if a[i]==4:
    count+=1
if count>=1:
  print(count)
else:
  print('not in array')