如何一致地打印timedelta（即格式化）

Question

我有一段代码，用于打印两个时间之间的毫秒差。但是，输出的时间格式并不符合我的期望，我希望得到一个统一且易读的格式。以下是原始代码及其输出：

#!/usr/bin/python
import datetime
import sys

date1 = datetime.datetime.strptime('20231107-08:52:53.539', '%Y%m%d-%H:%M:%S.%f')
date2 = datetime.datetime.strptime('20231107-08:52:53.537', '%Y%m%d-%H:%M:%S.%f')
diff = date1-date2
sys.stdout.write(str(diff) + '	')

date1 = datetime.datetime.strptime('20231107-08:52:53.537', '%Y%m%d-%H:%M:%S.%f')
date2 = datetime.datetime.strptime('20231107-08:52:53.537', '%Y%m%d-%H:%M:%S.%f')
diff = date1-date2
sys.stdout.write(str(diff) + '	')

date1 = datetime.datetime.strptime('20231107-08:52:53.532', '%Y%m%d-%H:%M:%S.%f')
date2 = datetime.datetime.strptime('20231107-08:52:53.537', '%Y%m%d-%H:%M:%S.%f')
diff = date1-date2
sys.stdout.write(str(diff) + '
')

其输出为：

$ python ./prntdatetest.py
0:00:00.002000	0:00:00	-1 day, 23:59:59.995000

而我期望的输出格式是：

00:00:00.002	00:00:00.000	-0:00:00.005

为了实现这个目标，并且坚持使用sys.stdout.write而非print函数，我应该如何调整代码呢？

ma11hew28 · Answer

您可以定义如下时间差格式化函数：

def format_diff(diff):
    符号 = '-' if diff < datetime.timedelta(0) else ''
    总秒数 = abs(diff.total_seconds())
    return f'{符号}{总秒数 // 3600 // 60:02.0f}:{总秒数 // 60 % 60:02.0f}:{总秒数 % 60:06.03f}'

然后在您的脚本中使用此函数来格式化输出：

#!/usr/bin/python
import datetime
import sys

date1 = datetime.datetime.strptime('20231107-08:52:53.539', '%Y%m%d-%H:%M:%S.%f')
date2 = datetime.datetime.strptime('20231107-08:52:53.537', '%Y%m%d-%H:%M:%S.%f')
diff = date1-date2
sys.stdout.write(format_diff(diff) + '	')

date1 = datetime.datetime.strptime('20231107-08:52:53.537', '%Y%m%d-%H:%M:%S.%f')
date2 = datetime.datetime.strptime('20231107-08:52:53.537', '%Y%m%d-%H:%M:%S.%f')
diff = date1-date2
sys.stdout.write(format_diff(diff) + '	')

date1 = datetime.datetime.strptime('20231107-08:52:53.532', '%Y%m%d-%H:%M:%S.%f')
date2 = datetime.datetime.strptime('20231107-08:52:53.537', '%Y%m%d-%H:%M:%S.%f')
diff = date1-date2
sys.stdout.write(format_diff(diff) + '
')

这样，您将获得期望的输出格式：

00:00:00.002    00:00:00.000    -00:00:00.005

注意：此方法适用于一天以内的时差处理。如果时差大于等于一天，您需要对函数进行相应的调整。

Lukasz Szozda · Answer

使用 diff.total_seconds() 方法可以从 timedelta 对象中获取绝对秒数。接着，通过取余和除法操作，可以分别得到小时、分钟和秒：

date1 = datetime.datetime.strptime('20231107-08:52:53.532', '%Y%m%d-%H:%M:%S.%f')
date2 = datetime.datetime.strptime('20231107-08:52:53.537', '%Y%m%d-%H:%M:%S.%f')
diff = date1 - date2

总秒数 = diff.total_seconds()
符号 = "-" if 总秒数 < 0 else ""
总秒数 = abs(总秒数)
sys.stdout.write(f"{符号}{总秒数 // 86400:.0f}:{总秒数 % 86400 // 3600:02.0f}:{总秒数 % 3600 // 60:02.0f}:{总秒数 % 60:06.03f}
")