Добавление хэшей из массива

Я пытаюсь создать массив / хэш из массива из нескольких хешей с одинаковыми ключами и средним значением. Мой массив:

[{:amount=>897500, :gross_amount=>897500, :tax=>147500, :hotel_fees=>0, :base_fare=>750000, :currency=>"INR"}, {:amount=>1006500, :gross_amount=>1006500, :tax=>156500, :hotel_fees=>0, :base_fare=>850000, :currency=>"INR"}]

Теперь я хочу вернуть что-то вроде этого:

{:amount=>952000, :gross_amount=>952000, :tax=>152000, :hotel_fees=>0, :base_fare=>800000, :currency=>"INR"}

где значения являются средними значениями от каждого хеша с тем же ключом.

Есть ли простой способ сделать это. Я попытался использовать слияние, но валюта становится 0 с ним.

Моя попытка:

p[0].merge(p[1]){|k,v1,v2| (v1+v2)/2 unless v1 && v2 == "INR"}

Редактировать:

На самом деле моя проблема на этом не закончилась, поэтому после получения среднего мне нужно было вставить значения в другой хеш. Поэтому я использовал что-то вроде этого:

        price_array = offer_values.map do |v| 
          v.inject do |k, v| 
            k.merge!(price: k[:price].merge(v[:price]){|_, a, b| [a, b].flatten })
          end
        end
        price_array.map do |o|
          o[:price] = {}.tap{ |h| o[:price].each {|k, list| h[k] = list.all?{|e| [Fixnum, NilClass].include? e.class} ? list.map(&:to_i).sum/list.size : list.compact.first ; h  } }
        end

Где offer_array - это тот, в котором мой оригинальный / первый массив находится в отдельных хешах. Это я пробовал с 2 и 3 хэшами, и это работает.

Если у вас есть предложения по улучшению кода, он открыт.

Источник

4 ответа

Решение

Для двух хешей в массиве вы можете использовать inject а также merge проверка, если значение для обоих currency ключи класса Fixnum, если нет, то взять значение валюты "INR" в первый хеш и используйте его:

array = [
  {:amount=>897500,  :gross_amount=>897500,  :tax=>147500, :hotel_fees=>0, :base_fare=>750000, :currency=>"INR"}, 
  {:amount=>1006500, :gross_amount=>1006500, :tax=>156500, :hotel_fees=>0, :base_fare=>850000, :currency=>"INR"}
]

p array.inject{|k,v| k.merge(v){|_,a,b| [a,b].all?{|e| e.is_a?(Fixnum)} ? (a+b)/2 : b}}
# => {:amount=>952000, :gross_amount=>952000, :tax=>152000, :hotel_fees=>0, :base_fare=>800000, :currency=>"INR"}

Для двух или более хэшей в массиве вы можете попробовать:

main_array = [
  {:amount=>897500,  :gross_amount=>897500,  :tax=>147500, :hotel_fees=>0, :base_fare=>750000, :currency=>"INR"}, 
  {:amount=>1006500, :gross_amount=>1006500, :tax=>156500, :hotel_fees=>0, :base_fare=>850000, :currency=>"INR"},
  {:amount=>1006500, :gross_amount=>1006500, :tax=>156500, :hotel_fees=>0, :base_fare=>850000, :currency=>"INR"},
]
array_result = main_array.flat_map(&:to_a).group_by(&:first).map do |key, array| 
  { 
    key => (
      result = array.inject(0) do |total, (_, value)| 
        value.is_a?(Fixnum) ? total + value : value
      end 
      result.is_a?(Fixnum) ? result / main_array.size : result 
    ) 
  } 
end
p array_result
Источник

IRB

2.2.3 :011 > b = {test1: 30, test2: 40}
 => {:test1=>30, :test2=>40} 
2.2.3 :012 > a = {test1: 20, test2: 60}
 => {:test1=>20, :test2=>60} 
2.2.3 :013 > c = a.merge(b){|key, oldval, newval| (newval + oldval)/2}
 => {:test1=>25, :test2=>50}
Источник

Это должно работать с любым количеством хэшей:

data = [
  { amount: 897_500, gross_amount: 897_500, tax: 147_500, hotel_fees: 0, base_fare: 750_000, currency: 'INR' },
  { amount: 1_006_500, gross_amount: 1_006_500, tax: 156_500, hotel_fees: 0, base_fare: 850_000, currency: 'INR' },
  { amount: 1_006_500, gross_amount: 1_006_500, tax: 156_500, hotel_fees: 0, base_fare: 850_000, currency: 'INR' }
]

transposed_hashes = data.each_with_object(Hash.new{|h, k| h[k] = []}) do |h, mem|
  h.each do |k, v|
    mem[k] << v
  end
end
# {:amount=>[897500, 1006500, 1006500], :gross_amount=>[897500, 1006500, 1006500], :tax=>[147500, 156500, 156500], :hotel_fees=>[0, 0, 0], :base_fare=>[750000, 850000, 850000], :currency=>["INR", "INR", "INR"]}

average_hash = transposed_hashes.map do |k, v|
  new_value = if v[0].is_a? Integer
                v.sum.to_f / v.size
              else
                v[0]
              end
  [k, new_value]
end.to_h

puts average_hash
# {:amount=>970166.6666666666, :gross_amount=>970166.6666666666, :tax=>153500.0, :hotel_fees=>0.0, :base_fare=>816666.6666666666, :currency=>"INR"}
Источник

Принятый ответ не будет работать для более чем 2 хешей, так как merge работает только 2 на 2, и вы рассчитываете среднее здесь.

(((3 + 2) / 2) + 2.5) / 2 is different from (3 + 2 + 2.5) / 3

Итак, я написал кусок кода, который может делать то, что вы хотите для любого размера массива, который у вас есть

  def self.merge_all_and_average(array)
    new_hash = {}
    unless array.empty?
      array[0].keys.each do |key|
        if array[0][key].class == Fixnum
          total = array.map { |i| i[key] }.inject(0) { |sum, x| sum + x }
          new_hash = new_hash.merge(key => total / array.size)
        else
          new_hash = new_hash.merge(key => array[0][key])
        end
      end
    end
    new_hash
  end
Источник
Другие вопросы по тегам