Como posso obter a próxima corda, na ordenação alfanumérica, em Python?
Pergunta
Eu preciso de um programa simples que, dada uma string, retorna para mim o próximo na ordenação alfanumérica (ou apenas a ordem alfabética).
f("aaa")="aab"
f("aaZ")="aba"
E assim por diante.
Existe uma função para este em um dos módulos já?
Solução
Eu não acho que há um built-in função para fazer isso. O seguinte deve funcionar:
def next_string(s):
strip_zs = s.rstrip('z')
if strip_zs:
return strip_zs[:-1] + chr(ord(strip_zs[-1]) + 1) + 'a' * (len(s) - len(strip_zs))
else:
return 'a' * (len(s) + 1)
Explicação: você encontra o último caractere que não é uma z
, incrementá-lo, e substituir todos os caracteres depois com a
de. Se a seqüência inteira é z
de, em seguida, retornar uma seqüência de todos os do a
que é um mais tempo.
Outras dicas
As respostas em Como você traduzir este de Perl para Python? suficiente? Não 100% o que você está pedindo, mas perto ...
Um diferente, mais longa, mas talvez mais legível e solução flexível:
def toval(s):
"""Converts an 'azz' string into a number"""
v = 0
for c in s.lower():
v = v * 26 + ord(c) - ord('a')
return v
def tostr(v, minlen=0):
"""Converts a number into 'azz' string"""
s = ''
while v or len(s) < minlen:
s = chr(ord('a') + v % 26) + s
v /= 26
return s
def next(s, minlen=0):
return tostr(toval(s) + 1, minlen)
s = ""
for i in range(100):
s = next(s, 5)
print s
Você converter a string em um número onde cada letra representa um dígito na base 26, aumentar o número por um e converter o número de volta para a string. Desta forma, você pode fazer matemática arbitrária em valores representados como sequências de letras.
O '' minLen '' parâmetro controla quantos dígitos o resultado terá (desde 0 == a == aaaaa).
suga que python não tem o que Ruby tem: String#next
Então aqui está uma solução de merda para lidar com cadeias alfa-numérica:
def next_string(s):
a1 = range(65, 91) # capital letters
a2 = range(97, 123) # letters
a3 = range(48, 58) # numbers
char = ord(s[-1])
for a in [a1, a2, a3]:
if char in a:
if char + 1 in a:
return s[:-1] + chr(char + 1)
else:
ns = next_string(s[:-1]) if s[:-1] else chr(a[0])
return ns + chr(a[0])
print next_string('abc') # abd
print next_string('123') # 124
print next_string('ABC') # ABD
# all together now
print next_string('a0') # a1
print next_string('1a') # 1b
print next_string('9A') # 9B
# with carry-over
print next_string('9') # 00
print next_string('z') # aa
print next_string('Z') # AA
# cascading carry-over
print next_string('a9') # b0
print next_string('0z') # 1a
print next_string('Z9') # AA0
print next_string('199') # 200
print next_string('azz') # baa
print next_string('Zz9') # AAa0
print next_string('$a') # $b
print next_string('$_') # None... fix it yourself
Não é ótimo. Meio que funciona para mim.