如何在列表理解中添加额外的中间步骤？

Question

假设我有一个输入。列出[str]对象，该对象包含“HH:mm”格式的时间戳，例如。

timestamps = ["22:58", "03:11", "12:21"]

我想把它转换成打字。列出[int]对象，并为每个时间戳列出“自午夜以来的分钟数”值：

converted = [22*60+58, 3*60+11, 12*60+21]

..., 但我想用一种风格，用一个列表来理解。我天真地构造了一个（语法上不正确的）实现

def timestamps_to_minutes(timestamps: typing.List[str]) -> typing.List[int]:
    return [int(hh) * 60 + int(mm) for ts in timestamps for hh, mm = ts.split(":")]

，但这不起作用，因为对于hh，mm=ts.split（“：”）不是有效的语法。。。

写同样的事情的有效方法是什么？

澄清一下：我可以看到一个形式上令人满意的解决方案

def timestamps_to_minutes(timestamps: typing.List[str]) -> typing.List[int]:
    return [int(ts.split(":")[0]) * 60 + int(ts.split(":")[1]) for ts in timestamps]

，但这是非常低效的，我不想把字符串拆分两次。

Answer 1

晚会迟到了。。但是为什么不使用datetime/timedelta来转换时间呢？

对于"hh： mm"，这可能有点夸张，但您可以轻松地将其调整到更复杂的时间字符串：

from datetime import datetime as dt
import typing

def timestamps_to_minutes(timestamps: typing.List[str]) -> typing.List[any]:
    """Uses datetime.strptime to parse a datetime string and return
    minutes spent in this day."""
    return [int(((p := dt.strptime(t,"%H:%M")) - dt(p.year,p.month, p.day)
                 ).total_seconds()//60) for t in timestamps]

timestamps = ["22:58", "03:11", "12:21"]

print(timestamps_to_minutes(timestamps))

产出：

[1378, 191, 741]

Answer 2

如果不想将字符串拆分两次，可以使用：=赋值运算符：

timestamps = [int((s := t.split(":"))[0]) * 60 + int(s[1]) for t in timestamps]
print(timestamps)

印刷品：

[1378, 191, 741]

备选方案：

print([int(h) * 60 + int(m) for h, m in (t.split(":") for t in timestamps)])

印刷品：

[1378, 191, 741]

Answer 3

可以使用内部生成器表达式进行拆分：

[int(hh)*60 + int(mm) for hh, mm in (ts.split(':') for ts in timestamps)]

虽然就我个人而言，我宁愿使用助手函数：

def timestamp_to_minutes(timestamp: str) -> int:
    hh, mm = timestamp.split(":")
    return int(hh)*60 + int(mm)

[timestamp_to_minutes(ts) for ts in timestamps]

# Alternative
list(map(timestamp_to_minutes, timestamps))