如何只对python循环中的第一项做某事？

Question

我想做一些与列表中的第一项不同的事情。这样做最pythonic的方式是什么？

for item in list:
    # only if its the first item, do something

    # otherwise do something else

Answer 1

几个选择，按 Pythonicity 降序排列：

for index, item in enumerate(lst): # note: don't use list
    if not index: # or if index == 0:
        # first item
    else:
        # other items

或者：

first = True
for item in lst:
    if first:
        first = False
        # first item 
    else:
        # other items 

或者：

for index in range(len(lst)):
    item = lst[i]
    if not index:
        # first item
    else:
        # other items

Answer 2

您可以使用 iter() 在列表上创建一个迭代器，然后对其调用 next() 以获取第一个值，然后循环其余的值。我发现这是处理文件的一种非常优雅的方式，其中第一行是标题，其余的是数据，即

list_iterator = iter(lst)

# consume the first item
first_item = next(list_iterator)

# now loop on the tail
for item in list_iterator:
    print(item)

Answer 3

do_something_with_first_item(lst[0])
for item in lst[1:]:
    do_something_else(item)

或者：

is_first_item = True
for item in lst:
    if is_first_item:
        do_something_with_first_item(item)
        is_first_item = False
    else:
        do_something_else(item)

不要使用list 作为变量名，因为这会影响内置函数list()。

jonrsharpe 的答案中基于enumerate 的解决方案优于此。您可能应该改用它。

Answer 4

使用处理第一个项目后更改的标志。例如：

first = True
for item in my_list:
    if first:
        # Processing unique to the first item
        first = False
    else:
        # Processing unique to other items
    # Shared processing

您也可以只处理第一项：

first = my_list.pop(0)
# Process first
for item in my_list:
    # Process item

# Add the first item back to the list at the beginning
my_list.insert(0, first)

Answer 5

jonrsharpe 使用enumerate 的第一个版本简洁明了，适用于所有可迭代对象：

for index, item in enumerate(lst):
    if not index:
        do_something_with_first_item(item)
    else:
        do_something_else(item)

senshin 使用lst[0] 和lst[1:] 的第一个解决方案更加简单，但仅适用于序列：

do_something_with_first_item(lst[0])
for item in lst[1:]:
    do_something_else(item)

您可以通过直接使用迭代器获得两全其美：

it = iter(lst)
do_something_with_first_item(next(it))
for item in it:
    do_something_else(item)

但是，尽管它是两全其美的，但实际上并没有那么简单。如果您知道自己有一个迭代器（因此您可以跳过第一行），或者您无论如何都需要一个迭代器来执行itertools 和genexpr 和类似的东西（所以你是已经要写第一行了）。在其他情况下是否值得做更多的是风格问题。