Как мне найти индекс максимального значения массива?
Я попробовал решение, рекомендованное здесь - в 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
как нули, а затем сначала преобразовать их в Float
s:
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