我有一个哈希数组:
arr = [ {:a => 1, :b => 2}, {:a => 3, :b => 4} ]
我想要实现的是:
arr.map{|x| x[:a]}.reduce(:+)
但我认为它有点难看,或者至少没有那么优雅:
arr.map(&:a).reduce(:+)
后一个是错误的,因为散列中没有调用方法a
。
有没有更好的写法map{|x| x[:a]}
?
您可以制作实际的对象,可能使用Struct:
MyClass = Struct.new :a, :b
arr = [MyClass.new(1, 2), MyClass.new(3, 4)]
arr.map(&:a).reduce(:+) #=> 4
或者为了获得更大的灵活性,一个OpenStruct:
require 'ostruct'
arr = [OpenStruct.new(a: 1, b: 2), OpenStruct.new(a: 3, b: 4)]
arr.map(&:a).reduce(:+) #=> 4
当然,这些中的任何一个都可以从现有的哈希中构造:
arr = [{ :a => 1, :b => 2 }, { :a => 3, :b => 4 }]
ss = arr.map { |h| h.values_at :a, :b }.map { |attrs| MyClass.new(*attrs) }
ss.map(&:a).reduce(:+) #=> 4
oss = arr.map { |attrs| OpenStruct.new attrs }
oss.map(&:a).reduce(:+) #=> 4
或者,对于更有创意、更实用的方法:
def hash_accessor attr; ->(hash) { hash[attr] }; end
arr = [{ :a => 1, :b => 2 }, { :a => 3, :b => 4 }]
arr.map(&hash_accessor(:a)).reduce(:+) #=> 4
目前尚不清楚您所说的“更好”是什么意思,以及为什么您认为正确的版本是丑陋的。
你喜欢这个“更好”吗?
arr.inject(0) { |sum, h| sum + h[:a] }