你可以使用 Ruby 的块简写来调用像数组访问器这样的方法吗？

Question

我习惯了可以缩短

some_array.map { |e| e.to_s }

到

some_array.map(&:to_s)

有没有办法缩短

some_array_of_arrays.map { |e| e[4] }

类似

some_array_of_arrays.map(&:[4])

显然我已经尝试了最后一个示例，但它不起作用。理想情况下，该解决方案将推广到其他“格式奇特”的方法调用，例如 []。

我不对任何 Rails/ActiveSupport 解决方案感兴趣。仅纯 Ruby，假设有某种解决方案。

Answer 1

你可以使用Proc：

> a = [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14]]
> third_elem = Proc.new {|x| x[2]}
> a.map(&third_elem)
#> [3, 7, 11, nil] 

或

> a.map &->(s) {s[2]}
#=> [3, 7, 11, nil] 

Answer 2

然后，您可以构建它。它没有那么优雅，但是......

class Call
  def self.[](name, *args)
    self.new(name, *args)
  end

  def initialize(name, *args)
    @proc = Proc.new do |obj|
      obj.send(name, *args)
    end
  end

  def to_proc
    @proc
  end
end

fourth = Call.new(:[], 3)
[[1,2,3,4,5],[6,7,8,9,10]].map(&fourth)           # => [4, 9]
# or equivalently
[[1,2,3,4,5],[6,7,8,9,10]].map(&Call.new(:[], 3)) # => [4, 9]
[[1,2,3,4,5],[6,7,8,9,10]].map(&Call[:[], 3])     # => [4, 9]

如果您想将其专门用于索引，您甚至可以简化为：

class Index
  def self.[](*args)
    self.new(*args)
  end

  def initialize(*args)
    @proc = Proc.new do |obj|
      obj[*args]
    end
  end

  def to_proc
    @proc
  end
end

[[1,2,3,4,5],[6,7,8,9,10]].map(&Index[3])     # => [4, 9]

或者，更短的，正如@muistooshort 在 cmets 中展示的那样，如果你不想有一个专门的完整课程：

index = ->(*ns) { ->(a) { a[*ns] } }
[[1,2,3,4,5],[6,7,8,9,10]].map(&index[3])     # => [4, 9]