Как мне найти индекс максимального значения массива?

Я попробовал решение, рекомендованное здесь - в Ruby, какой самый чистый способ получения индекса наибольшего значения в массиве?

array = [nil, nil, nil, nil, nil, 0.9655172413793104, nil, nil]
idx = array.each_with_index.max[1]

Но я получаю некоторые исключения:

ArgumentError: comparison of Array with Array failed
    from (irb):4:in `each'
    from (irb):4:in `each_with_index'
    from (irb):4:in `each'
    from (irb):4:in `max'
    from (irb):4
    from /Users/davea/.rvm/gems/ruby-2.4.0/gems/railties-5.0.3/lib/rails/commands/console.rb:65:in `start'
    from /Users/davea/.rvm/gems/ruby-2.4.0/gems/railties-5.0.3/lib/rails/commands/console_helper.rb:9:in `start'
    from /Users/davea/.rvm/gems/ruby-2.4.0/gems/railties-5.0.3/lib/rails/commands/commands_tasks.rb:78:in `console'
    from /Users/davea/.rvm/gems/ruby-2.4.0/gems/railties-5.0.3/lib/rails/commands/commands_tasks.rb:49:in `run_command!'
    from /Users/davea/.rvm/gems/ruby-2.4.0/gems/railties-5.0.3/lib/rails/commands.rb:18:in `<top (required)>'
    from bin/rails:4:in `require'
    from bin/rails:4:in `<main>'

2 ответа

Решение

Если вы хотите опустить nilИсходя из результата, вы можете использовать:

array.index(array.compact.max)

Или если вы хотите лечить nilкак нули, а затем сначала преобразовать их в Floats:

array.index(array.map(&:to_f).max)

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

def max_idx(array)
  mx = array.select { |e| e.kind_of?(Numeric) }.max 
  mx ? array.each_index.select { |i| array[i] == mx } : nil
end      

require 'bigdecimal'
max_idx [nil,3,1]                   #=> [1]
max_idx [nil,3.2,"cat",1]           #=> [1]
max_idx [nil,3,nil,1,3]             #=> [1,4]
max_idx [nil,3.2,nil,1,3.2]         #=> [1,4]
max_idx [nil,Rational(3),1]         #=> [1]
max_idx [nil,BigDecimal.new("3"),1] #=> [1]
max_idx [nil,nil,nil]               #=> nil
Другие вопросы по тегам