Что такое Ruby-эквивалент Python itertools, особенно Комбинации / перестановки / GroupBy? - PullRequest
13 голосов
/ 14 марта 2010

Модуль Python itertools предоставляет множество полезных возможностей в отношении обработки итерируемого / итератора с использованием генераторов. Например,

permutations(range(3)) --> 012 021 102 120 201 210

combinations('ABCD', 2) --> AB AC AD BC BD CD

[list(g) for k, g in groupby('AAAABBBCCD')] --> AAAA BBB CC D

Что такое эквивалент в Ruby?

Под эквивалентом я подразумеваю быструю и эффективную память (модуль Python itertools написан на C).

1 Ответ

18 голосов
/ 14 марта 2010

Array#permutation, Array#combination и Enumerable#group_by определены в рубине начиная с 1.8.7. Если вы используете 1.8.6, вы можете получить эквивалентные методы из фасетов или active_support или backports .

Пример использования:

[0,1,2].permutation.to_a
#=> [[0, 1, 2], [0, 2, 1], [1, 0, 2], [1, 2, 0], [2, 0, 1], [2, 1, 0]]

[0,1,2,3].combination(2).to_a
#=> [[0, 1], [0, 2], [0, 3], [1, 2], [1, 3], [2, 3]]

[0,0,0,1,1,2].group_by {|x| x}.map {|k,v| v}
#=> [[0, 0, 0], [1, 1], [2]]

[0,1,2,3].group_by {|x| x%2}
#=> {0=>[0, 2], 1=>[1, 3]}
...