рубин значение уник и сумма

Question

У меня есть массив массивов

arr = [ 
    ["BEER", 37], 
    ["BEER", 95], 
    ["BEER", 85], 
    ["BEER", 60], 
    ["BEER", 36], 

    ["FOOD", 31], 
    ["FOOD", 86], 
    ["FOOD", 50], 
    ["FOOD", 0] 
] 

Как извлечь массив как этот

new_arr = [ [ 'FOOD', sum_of_food ], 
      ['BEER', sum_of_beer] ] 

Я только что получил arr по Model.all.map {|c| [c.source, c.amount ] }

Answer 1

Вы можете использовать Enumerable#inject в цикле каждого элемент в коллекции и вычислить отстойник в новый массив, который используется в качестве аккумулятора.

arr = [ 
["BEER", 37], 
["BEER", 95], 
["BEER", 85], 
["BEER", 60], 
["BEER", 36], 

["FOOD", 31], 
["FOOD", 86], 
["FOOD", 50], 
["FOOD", 0], 
] 

arr.inject({}) do |accumulator, (what, total)| 
    accumulator[what] ||= 0 
    accumulator[what] += total 
    accumulator 
end 

Если вам нужен Array вместо Hash, просто вызовите to_a на результат.

---

Если данные хранятся в базе данных, вы можете запросить непосредственно в базе данных с помощью оператора SQL SUM().

results = Model.select('source, SUM(amount) AS total').group(:source) 
result = results.first 
result.source 
# => FOO 
result.total 
# => the sum of amount for FOO 

Answer 2

Вы можете использовать три Enumerable методы: group_by, map и reduce (ака inject):

arr.group_by(&:first).map { |k,v| [k, v.map(&:last).reduce(:+)] } 
    #=> [["BEER", 313],["FOOD", 167]] 

стадии:

h = arr.group_by(&:first) 
    # => {"BEER"=>[["BEER", 37], ["BEER", 95], ["BEER", 85], ["BEER", 60], 
    #    ["BEER", 36]], 
    #  "FOOD"=>[["FOOD", 31], ["FOOD", 86], ["FOOD", 50], ["FOOD", 0]]} 

map присвоит блочные переменные первого элемент (значение ключа pa л) из h:

k = "BEER" 
v = [["BEER", 37], ["BEER", 95], ["BEER", 85], ["BEER", 60], ["BEER", 36]] 

затем выполняет вычисление блока:

[k, v.map(&:last).reduce(:+)] 
    #=> ["BEER", [37, 95, 85, 60, 36].reduce(:+)] 
    # ["BEER", 313] 

Второй элемент h передается к блоку:

k = "FOOD" 
v = [["FOOD", 31], ["FOOD", 86], ["FOOD", 50], ["FOOD", 0]] 
[k, v.map(&:last).reduce(:+)] 
    #=> ["FOOD", [31, 86, 50, 0].reduce(:+)] 
    # ["FOOD", 167] 

Если пары arr известны которые должны быть сгруппированы первым элементом пары, как они приведены в примере, мы могли бы использовать Enumerable#chunk:

arr.chunk(&:first).map { |k,v| [k, v.map(&:last).reduce(:+)] } 

^{1 Поскольку ключи хеширования Ruby v1.9 заказываются по их порядку размещения. Поэтому каждый может ссылаться на n-й элемент (пару-ключ-значение) хэша. До v1.9 не было понято, какие ключи хеша заказываются. Для этих версий приведенный выше код работает нормально, но мы не знаем, какой порядок пар ключ-значение передается map его блоку.}