将字符串重复到一定长度的有效方法是什么?例:repeat('abc', 7) -> 'abcabca'

这是我当前的代码:

def repeat(string, length):
    cur, old = 1, string
    while len(string) < length:
        string += old[cur-1]
        cur = (cur+1)%len(old)
    return string

有没有更好的(更python化的)方法来做到这一点?也许使用列表理解?


当前回答

Jason Scheirer的答案是正确的,但需要更多的阐述。

首先,要将一个字符串重复整数次,你可以使用重载乘法:

>>> 'abc' * 7
'abcabcabcabcabcabcabc'

所以,为了重复一个字符串,直到它至少和你想要的长度一样长,你计算出适当的重复次数,并把它放在乘法运算符的右边:

def repeat_to_at_least_length(s, wanted):
    return s * (wanted//len(s) + 1)

>>> repeat_to_at_least_length('abc', 7)
'abcabcabc'

然后,你可以用数组切片把它修剪成你想要的长度:

def repeat_to_length(s, wanted):
    return (s * (wanted//len(s) + 1))[:wanted]

>>> repeat_to_length('abc', 7)
'abcabca'

或者,正如在pillmod的回答中所建议的那样,可能没有人会再向下滚动到足够远的地方,你可以使用divmod来计算所需的完整重复的数量,以及额外的字符数量,一次全部:

def pillmod_repeat_to_length(s, wanted):
    a, b = divmod(wanted, len(s))
    return s * a + s[:b]

哪个更好?让我们进行基准测试:

>>> import timeit
>>> timeit.repeat('scheirer_repeat_to_length("abcdefg", 129)', globals=globals())
[0.3964178159367293, 0.32557755894958973, 0.32851039397064596]
>>> timeit.repeat('pillmod_repeat_to_length("abcdefg", 129)', globals=globals())
[0.5276265419088304, 0.46511475392617285, 0.46291469305288047]

所以,pillmod的版本大约慢了40%,这太糟糕了,因为我个人认为它的可读性更高。这有几个可能的原因,首先是它多编译了大约40%的字节码指令。

注意:这些例子使用new-ish //运算符来截断整数除法。这通常被称为Python 3特性,但根据PEP 238,它是在Python 2.2中引入的。你只需要在python3中使用它(或在有__future__导入除法的模块中),但你可以无论如何使用它。

其他回答

def extended_string (word, length) :

    extra_long_word = word * (length//len(word) + 1)
    required_string = extra_long_word[:length]
    return required_string

print(extended_string("abc", 7))
def repeat_to_length(string_to_expand, length):
   return (string_to_expand * ((length/len(string_to_expand))+1))[:length]

python3:

def repeat_to_length(string_to_expand, length):
    return (string_to_expand * (int(length/len(string_to_expand))+1))[:length]

Jason Scheirer的答案是正确的,但需要更多的阐述。

首先,要将一个字符串重复整数次,你可以使用重载乘法:

>>> 'abc' * 7
'abcabcabcabcabcabcabc'

所以,为了重复一个字符串,直到它至少和你想要的长度一样长,你计算出适当的重复次数,并把它放在乘法运算符的右边:

def repeat_to_at_least_length(s, wanted):
    return s * (wanted//len(s) + 1)

>>> repeat_to_at_least_length('abc', 7)
'abcabcabc'

然后,你可以用数组切片把它修剪成你想要的长度:

def repeat_to_length(s, wanted):
    return (s * (wanted//len(s) + 1))[:wanted]

>>> repeat_to_length('abc', 7)
'abcabca'

或者,正如在pillmod的回答中所建议的那样,可能没有人会再向下滚动到足够远的地方,你可以使用divmod来计算所需的完整重复的数量,以及额外的字符数量,一次全部:

def pillmod_repeat_to_length(s, wanted):
    a, b = divmod(wanted, len(s))
    return s * a + s[:b]

哪个更好?让我们进行基准测试:

>>> import timeit
>>> timeit.repeat('scheirer_repeat_to_length("abcdefg", 129)', globals=globals())
[0.3964178159367293, 0.32557755894958973, 0.32851039397064596]
>>> timeit.repeat('pillmod_repeat_to_length("abcdefg", 129)', globals=globals())
[0.5276265419088304, 0.46511475392617285, 0.46291469305288047]

所以,pillmod的版本大约慢了40%,这太糟糕了,因为我个人认为它的可读性更高。这有几个可能的原因,首先是它多编译了大约40%的字节码指令。

注意:这些例子使用new-ish //运算符来截断整数除法。这通常被称为Python 3特性,但根据PEP 238,它是在Python 2.2中引入的。你只需要在python3中使用它(或在有__future__导入除法的模块中),但你可以无论如何使用它。

耶递归!

def trunc(s,l):
    if l > 0:
        return s[:l] + trunc(s, l - len(s))
    return ''

不会永远缩放,但对于较小的字符串是可以的。它很漂亮。

我承认我刚读了《小阴谋家》我现在喜欢递归。

另一种FP方法:

def repeat_string(string_to_repeat, repetitions):
    return ''.join([ string_to_repeat for n in range(repetitions)])