更快的strptime？

Question

我有读取'YYYY-MM-DD‘格式的大量日期的代码。解析所有这些日期，以便它可以添加一天、两天或三天，然后以相同的格式写回，这会大大降低速度。

 3214657   14.330    0.000  103.698    0.000 trade.py:56(effective)
 3218418   34.757    0.000   66.155    0.000 _strptime.py:295(_strptime)

 day = datetime.datetime.strptime(endofdaydate, "%Y-%m-%d").date()

有什么建议可以让它更快一点(或者更快一点)吗？

Answer 1

Python 3.7+：fromisoformat()

从Python3.7开始，datetime类就有了fromisoformat方法。应该注意的是，这也可以应用于这个问题：

性能与strptime()

与普通的fromisoformat相比，显式字符串切分可能会使性能提高约9倍，但使用内置的方法可以将性能提高约90倍！

%timeit isofmt(datelist)
569 µs ± 8.45 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

%timeit slice2int(datelist)
5.51 ms ± 48.9 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

%timeit normalstrptime(datelist)
52.1 ms ± 1.27 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)

from datetime import datetime, timedelta
base, n = datetime(2000, 1, 1, 1, 2, 3, 420001), 10000
datelist = [(base + timedelta(days=i)).strftime('%Y-%m-%d') for i in range(n)]

def isofmt(l):
    return list(map(datetime.fromisoformat, l))
    
def slice2int(l):   
    def slicer(t):
        return datetime(int(t[:4]), int(t[5:7]), int(t[8:10]))
    return list(map(slicer, l))

def normalstrptime(l):
    return [datetime.strptime(t, '%Y-%m-%d') for t in l]
    
print(isofmt(datelist[0:1]))
print(slice2int(datelist[0:1]))
print(normalstrptime(datelist[0:1]))

# [datetime.datetime(2000, 1, 1, 0, 0)]
# [datetime.datetime(2000, 1, 1, 0, 0)]
# [datetime.datetime(2000, 1, 1, 0, 0)]

Python 3.8.3rc1 x64 / Win10

Answer 2

因子7 lot足够吗？

datetime.datetime.strptime(a, '%Y-%m-%d').date()       # 8.87us

datetime.date(*map(int, a.split('-')))                 # 1.28us

编辑：显式切片的好主意：

datetime.date(int(a[:4]), int(a[5:7]), int(a[8:10]))   # 1.06us

这就是因子8。

Answer 3

对于ISO格式的非时区字符串，例如："2021-01-04T14:30:03.123"

datetime.datetime(int(d[:4]), int(d[5:7]), int(d[8:10]), int(d[11:13]), int(d[14:16]), int(d[17:19]), int(d[20:]))

似乎比strptime()和fromisoformat()运行得更快。