python循环和字符串操作

Question

函数名为def expand_fmla（原创）

这个函数的输入变量 original 是一个具有特定格式的字符串：original 的前两个位置有符号* 或 +，因此 original 的前两个位置有 4 种可能： ++, **, +* 和 *+。后面的位置有数字，至少有 3 个（0 到 9），可能包括重复。

这个函数应该返回一个与原始公式具有相同数字和顺序的公式，并且在数字之间交替包含两个操作符号。

例如：

expand_fmla('++123') should return '1+2+3'

expand_fmla('+*1234') should return '1+2*3+4'

expand_fmla('*+123456') should return '1*2+3*4+5*6'

我怎么能这样，我不明白？？？

Answer 1

应该这样做：

def expand_fmla(original):
    answer = []
    ops = original[:2]
    nums = original[2:]
    for i,num in enumerate(nums):
        answer.extend([num, ops[i%2]])
    return ''.join(answer[:-1])

In [119]: expand_fmla('+*1234')
Out[119]: '1+2*3+4'

In [120]: expand_fmla('*+123456')
Out[120]: '1*2+3*4+5*6'

In [121]: expand_fmla('++123')
Out[121]: '1+2+3'

希望对你有帮助

Answer 2

使用一些 itertools 食谱，特别是 itertools.cycle() 在这里会很有用：

from itertools import *
def func(x):
    op=x[:2]                             # operators
    opn=x[2:]                            # operands
    cycl=cycle(op)                       # create a cycle iterator of operators

    slice_cycl=islice(cycl,len(opn)-1)   # the no. of times operators are needed
                                         # is 1 less than no. of operands, so
                                         # use islice() to slice the iterator

    # iterate over the two iterables opn and          
    # slice_cycl simultanesouly using
    # izip_longest , and as the length of slice_cycl 
    # is 1 less than len(opn), so we need to use a fillvalue=""
    # for the last pair of items

    lis=[(x,y) for x,y in izip_longest(opn,slice_cycl,fillvalue="")]   


    return "".join(chain(*lis))

print func("++123")
print func("+*123")
print func("+*123456")
print func("*+123456")

输出：

1+2+3
1+2*3
1+2*3+4*5+6
1*2+3*4+5*6