遍历列表

Question

我对 Python 很陌生，因此这个问题。

我有一个代表日期的列表，即三月的星期一和四月初

[2, 9, 16, 23, 30, 6]

列表“color_sack”是从我们当地议会网站的抓取中创建的。

我正在使用

next_rubbish_day = next(x for x in color_sack if x > todays_date.day)

todays_date.day 只返回代表日期的数字，即 30

这整个月都运行良好，直到今天 30 号，现在显示错误

next_rubbish_day = next(x for x in color_sack if x > todays_date.day)
StopIteration

是否有可能以更好的方式逐步浏览列表，以便 next_rubbish_day 填充上面列表中 30 之后的 6。 我可以理解为什么它不起作用但无法找到更好的方法。

4 月开始时，列表将更新为 4 月到 5 月初的星期一的新日期

Answer 1

next接受一个可选的默认值，当可迭代对象为空时返回该默认值。 如果color_sack始终在最后一个位置具有下个月的第一天，则将其作为默认值返回：

next_rubbish_day = next(
    (x for x in color_sack[:-1] if x > todays_date.day),
    color_sack[-1],
)

请注意，此方案不会告诉您是否翻车。 它只会告诉您下一个日期是 6 日，而不是 4 月 6 日和 3 月 6 日。

为避免使用魔法索引，请考虑明确拆分列表并为每个部分指定适当的名称。

*this_month, fallback_day = color_sack
next_rubbish_day = next(
    (day for day in this_month if day > todays_date.day),
    fallback_day,
)

---

如果您需要了解月份，请明确处理StopIteration ：

try:
    day = next(x for x in color_sack[:-1] if x > todays_date.day)
except StopIteration:
    day = color_sack[-1]
    month = 'next'
else:
    month = 'this'
print(f'Next date is {day} of {month} month')

Answer 2

考虑一下，如果您当前的月份是三月，相应的日期列表是[2, 9, 16, 23, 30, 6]并且今天的日期是30 ，那么我们所做的基本上是：

1. 检查color_sack中是否有大于
   如果是今天的日期，那么我们产生那个日期。 在我们的例子中，列表中没有日期大于30 。
2. 如果第1st条件失败，我们现在找出color_sack中最大日期的color_sack ，在我们的例子中，最大日期是30 ，它的索引是4 ，现在我们发现是否存在大于最大日期索引的idx 。列表，如果是，则返回该日期。

该算法将遵守当前月份中的任何日期，例如March 。 一旦新的一个月开始，例如。 “从 4 月开始，列表将更新为 4 月到 5 月初的星期一的新日期”。 所以这个算法将始终遵守。

尝试这个：

def next_rubbish_day(color_sack, todays_date):
    for idx, day in enumerate(color_sack):
        if day > todays_date or idx > color_sack.index(max(color_sack)):
            yield day

---

print(next(next_rubbish_day(color_sack, 6)))
print(next(next_rubbish_day(color_sack, 10)))
print(next(next_rubbish_day(color_sack, 21)))
print(next(next_rubbish_day(color_sack, 30)))
print(next(next_rubbish_day(color_sack, 31)))

---

输出：

9
16
23
6
6

Answer 3

感谢您的帮助，我使用了 MisterMiyagi 片段，因为它目前似乎有效。

这是完整的代码：

import datetime
import requests
import calendar
from bs4 import BeautifulSoup
from datetime import date


def ord(n):  # returns st, nd, rd and th
    return str(n) + (
        "th" if 4 <= n % 100 <= 20 else {
            1: "st", 2: "nd", 3: "rd"}.get(n % 10, "th")
    )


# Scrapes rubbish collection dates
URL = "https://apps.castlepoint.gov.uk/cpapps/index.cfm?roadID=2767&fa=wastecalendar.displayDetails"
raw_html = requests.get(URL)
data = BeautifulSoup(raw_html.text, "html.parser")

pink = data.find_all('td', class_='pink', limit=3)
black = data.find_all('td', class_='normal', limit=3)
month = data.find('div', class_='calMonthCurrent')

# converts .text and strip [] to get month name
month = str((month.text).strip('[]'))

todays_date = datetime.date.today()
print()

# creats sack lists
pink_sack = []
for div in pink:
    n = div.text
    pink_sack.append(n)
pink_sack = list(map(int, pink_sack))
print(f"Pink list {pink_sack}")

black_sack = []
for div in black:
    n = div.text
    black_sack.append(n)
black_sack = list(map(int, black_sack))
print(f"Black list {black_sack}")

# creats pink/black list
color_sack = []
color_sack = [None]*(len(pink_sack)+len(black_sack))
color_sack[::2] = pink_sack
color_sack[1::2] = black_sack
print(f"Combined list {color_sack}")
print()
print()

# checks today for rubbish
if todays_date.day in color_sack:
    print(f"Today {(ord(todays_date.day))}", end=" ")
if todays_date.day in pink_sack:
    print("is pink")
elif todays_date.day in black_sack:
    print("is black")

# Looks for the next rubbish day
next_rubbish_day = next(
    (x for x in color_sack[:-1] if x > todays_date.day),
    color_sack[-1],
)

# gets day number
day = calendar.weekday(
    (todays_date.year), (todays_date.month), (next_rubbish_day))


# print(next_rubbish_day)
print(f"Next rubbish day is {(calendar.day_name[day])} the {(ord(next_rubbish_day))}" +
      (" and is Pink" if next_rubbish_day in pink_sack else " and is Black"))
print()

可能有很多更有效的方法可以做到这一点，所以我愿意接受建议并一直在学习。