Pythonic maneira a vírgula divisão separada números em pares
Pergunta
Eu gostaria de dividir um valor separado por vírgula em pares:
>>> s = '0,1,2,3,4,5,6,7,8,9'
>>> pairs = # something pythonic
>>> pairs
[(0, 1), (2, 3), (4, 5), (6, 7), (8, 9)]
O que seria # algo pythônico Look Like?
Como você detectar e tratar uma string com um conjunto ímpar de números?
Solução
Algo como:
zip(t[::2], t[1::2])
exemplo completa:
>>> s = ','.join(str(i) for i in range(10))
>>> s
'0,1,2,3,4,5,6,7,8,9'
>>> t = [int(i) for i in s.split(',')]
>>> t
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> p = zip(t[::2], t[1::2])
>>> p
[(0, 1), (2, 3), (4, 5), (6, 7), (8, 9)]
>>>
Se o número de itens é estranho, o último elemento será ignorado. Apenas pares completos serão incluídos.
Outras dicas
A opção mais geral, que também funciona em iterators e permite combinar qualquer número de itens:
def n_wise(seq, n):
return zip(*([iter(seq)]*n))
Substituir zip com itertools.izip se você deseja obter um iterador preguiçoso em vez de uma lista.
Como sobre isto:
>>> x = '0,1,2,3,4,5,6,7,8,9'.split(',')
>>> def chunker(seq, size):
... return (tuple(seq[pos:pos + size]) for pos in xrange(0, len(seq), size))
...
>>> list(chunker(x, 2))
[('0', '1'), ('2', '3'), ('4', '5'), ('6', '7'), ('8', '9')]
Isto irá também muito bem lidar com quantidades desiguais:
>>> x = '0,1,2,3,4,5,6,7,8,9,10'.split(',')
>>> list(chunker(x, 2))
[('0', '1'), ('2', '3'), ('4', '5'), ('6', '7'), ('8', '9'), ('10',)]
P.S. Eu tive esse código escondidos e eu só percebi onde eu tenho que partir. Há duas perguntas muito semelhantes em stackoverflow sobre isso:
- Qual é o mais maneira “pythônico” para iterar sobre uma lista em pedaços?
- Como você dividir um lista em pedaços de tamanho uniforme em Python?
Há também esta jóia da seção Receitas de itertools
:
def grouper(n, iterable, fillvalue=None):
"grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx"
args = [iter(iterable)] * n
return izip_longest(fillvalue=fillvalue, *args)
Uma solução muito parecido FogleBirds, mas usando um iterador (uma expressão gerador) em vez de compreensão da lista.
s = '0,1,2,3,4,5,6,7,8,9'
# generator expression creating an iterator yielding numbers
iterator = (int(i) for i in s.split(','))
# use zip to create pairs
# (will ignore last item if odd number of items)
# Note that zip() returns a list in Python 2.x,
# in Python 3 it returns an iterator
pairs = zip(iterator, iterator)
Ambas as compreensões de lista e gerador de expressões provavelmente seria considerado bastante "pythônico".
Isto irá ignorar o último número em uma lista de estranho:
n = [int(x) for x in s.split(',')]
print zip(n[::2], n[1::2])
Esta vontade pad a lista mais curta por 0 em uma lista de estranho:
import itertools
n = [int(x) for x in s.split(',')]
print list(itertools.izip_longest(n[::2], n[1::2], fillvalue=0))
izip_longest está disponível em Python 2.6.