如何在 python 中为我的范围函数设置限制？

Question

假设我正在遍历一个列表。 我想检查列表的邻居（i-1 和 i+1）是否包含某个元素。 如何在不遇到“列表索引超出范围”问题的情况下执行此操作？ 例子：

list = [1, 0, 1, 0, 1, 0]
for i, j in enumerate(list):
    elements = 0
    for m in range(i-1,i+2):
        if list[m] == 1:
            elements += 1
    print(list[i], elements)

如何为 range 函数设置边界，使其不会低于 0 和高于 len(list)？

Answer 1

尝试从顶部和底部切片列表

list = [1, 0, 1, 0, 1, 0]
elements = 0
# slice list and start from second element and finish at the penultimate element
for i, j in enumerate(list[1:-1], 1):
    for m in range(i-1,i+2):
        if list[m] == 1:
            elements += 1

或者由于您不在外部循环中使用列表项，因此循环范围

elements = 0
# start from the second index and finish at the penultimate index
for i in range(1, len(list)-1):
    for m in range(i-1,i+2):
        if list[m] == 1:
            elements += 1

Answer 2

如果要迭代目标列表中的所有元素，一种解决方案是检查第二个 for 循环的值：

_list = [1, 0, 1, 0, 1, 0]
elements = 0
for i, j in enumerate(_list):
    for m in range(max(i-1, 0), min(i+2, len(_list))):
        if _list[m] == 1:
            elements += 1

Answer 3

听起来您想使用窗口功能。 我在这里的某个地方得到了这个，并且多年来一直在使用它：

from typing import Generator
from itertools import islice

def window(seq, n: int = 2) -> Generator:
    """
    Returns a sliding window (of width n) over data from the iterable
    """
    it = iter(seq)
    result = tuple(islice(it, n))
    if len(result) == n:
        yield result
    for elem in it:
        result = result[1:] + (elem,)
        yield result
        

mylist = [1, 0, 1, 0, 1, 0, 5]
for chunk in window(mylist, n=3):
    print(chunk)

这会给你：

(1, 0, 1) (0, 1, 0) (1, 0, 1) (0, 1, 0) (1, 0, 5)

您可以随意比较生成的“窗口”的内容。