2012-07-31 8 views
7

Próbuję zgrupować tablicę liczb całkowitych w haszowanie na podstawie tego, gdzie poszczególne wartości mieszczą się w zakresie. Zasadniczo chcę przekonwertować tablicę na histogram o stałej szerokości.Ruby Array to Histogram: jak grupować liczby według zakresu?

przykład:

values = [1,3,4,4,4,4,4,10,12,15,18] 
bin_width = 3 

muszę grupy wartości tablicy w histogram zakresie oparte o, gdzie wchodzą one w 3-jednostkę szerokości wiadra tak:

{'0..2'=>[1,3],'3..5'=>[4,4,4,4,4],'6..8'=>[],'9..11'=>[10].... 

jest istnieje proste rozwiązanie jednoliniowe (może coś takiego jak values.group_by{|x| #range calc}), które działałoby tutaj?

+1

pytania: 1) Chyba 3 powinny być 3..5. 2) po co używać ciągów jako kluczy zamiast prawdziwych zakresów ?, 3) potrzebujesz również pustych zakresów? – tokland

+3

Nie powinno być obsesji na punkcie pisania jednoosobowych. Dążenie do rozwiązań, które obejmują tylko wyrażenia (to jest to, stosując podejście funkcjonalne) tak, obawiając się przydziałów, aby zapisać kilka linii, nie. – tokland

Odpowiedz

10
values = [1, 7, 2, 8, 2] 
values.group_by { |x| x/3 }.map { |k, vs| [(3*k..3*k+2), vs] }.to_h 
#=> {0..2=>[1, 2, 2], 6..8=>[7, 8]} 

Jeśli naprawdę potrzebujesz pustych zakresów, nie sądzę, aby była możliwa czysta jednolinijka. Ale powinien to zrobić:

grouped = values.group_by { |x| x/3 } 
min, max = grouped.keys.minmax 
(min..max).map { |n| [(3*n..3*n+2), grouped.fetch(n, [])] }.to_h 
#=> {0..2=>[1, 2, 2], 3..5=>[], 6..8=>[7, 8]} 
+0

Hah, chciałem tylko zasugerować coś, co nie wymaga Facetów, a następnie zaktualizowałeś swój post. –

+1

@Michael, tak, przepraszam, w rzeczywistości mój fragment aspektu był całkowicie błędny, map_by nie jest tutaj użyteczny, musimy przetworzyć klucze, a nie wartości. Tak to napisałeś? – tokland

4

wymyśliłem raczej nieefektywne, ale dość klarownego roztworu:

ranges = 0.step(values.max, bin_width).each_cons(2).map { |s, e| Range.new(s, e, true) } 
values.group_by { |v| ranges.find { |r| r.cover? v } } 
+0

Dzięki za bardziej ogólną wersję, która działa zarówno z 3-jednostkowymi koszami i może dostosować się do bardziej przypadkowych przypadków. Dokładnie to, czego potrzebowałem. – slothbear