Python解析字符串，已知结构

Question

我必须用已知的结构解析一个简单的字符串列表，但是我发现它不必要地笨拙。 我觉得我错过了一个诀窍，也许是一些简单的正则表达式可以让这件事变得微不足道？

这个字符串指的是未来的年/月数，我想把它变成十进制年份。

通用格式：“aYbM”

其中a是年数，b是月数，可以是int，两者都是可选的（以及它们的标识符）

测试用例：

5Y3M == 5.25
5Y == 5.0
6M == 0.5
10Y11M = 10.91666..
3Y14M = raise ValueError("string '%s' cannot be parsed" %input_string)

到目前为止，我的尝试涉及到字符串拆分，虽然它们确实产生了正确的结果，但还是相当麻烦：

^{pr2}$

Answer 1

您可以使用re.findall

>>> def parse(m):
    s = 0
    j = re.findall(r'\d+Y|\d+M', m)
    for i in j:
        if 'Y' in i:
            s += float(i[:-1])
        if 'M' in i:
            s += float(i[:-1])/12
    print(s)


>>> parse('5Y')
5.0
>>> parse('6M')
0.5
>>> parse('10Y11M')
10.916666666666666
>>> parse('3Y14M')
4.166666666666667