[ruby-dev:42741] [Ruby 1.9-Feature#4147] Array#sample で重みを指定したい

チケット #4147 が更新されました。 (by Yoji Ojima)


メソッドの引数を考慮しない簡易的なコードでよろしければ。

class Array
  alias :org_sample :sample

  def sample
    if block_given?
      prob_map = collect {|v| yield(v) }
      rn = rand * prob_map.inject(0) {|r, v| r += v }
      prob_map.each_with_index do |v, i|
        rn -= v
        return self[i] if rn < 0
      end
    else
      org_sample
    end
  end
end
----------------------------------------
http://redmine.ruby-lang.org/issues/show/4147

----------------------------------------
http://redmine.ruby-lang.org