Итерация для выбора и изменения в рубин

Question

У меня есть код, похожий на

@a.each{ |n| 
    if n == i then 
     @b.push n 
     n = 2*n 
    end 
} 

Я хочу, чтобы перебрать массив @a, и если какое-то условие n == i верно, то использовать текущий элемент n сделать @b.push n, и затем обновите его до n = 2*n на месте. Как я могу выполнять итерацию для выбора и изменения?

Answer 1

@a.each.with_index do |n, j| 
    next unless n == i 
    @b.push(i) 
    @a[j] = 2 * i 
end 

Answer 2

@a.each_with_index do |n, x| 
    if n == i 
    @b.push n 
    @a[x] = 2*n 
    end 
end 

Answer 3

использованием each_with_index решена проблема

@a.each_with_index{ |n,q| 
    if n == i then 
     @b.push n 
     @a[q] = 2*n 
    end 
} 

Answer 4

NArray может сделать это очень быстро, если массив велик:

require 'narray' 

# Some sample data: 
a = NArray.int(100000).random(10) 

i = 8 
# b is a "boolean" array that can be used to select or modify 
# only certain elements in a 
b = a.eq i 

a[b] *= 2 

Если ваш первоначальный массив @a велик, и вы беспокоятся о производительности, возможно, вы захотите попробовать.

Answer 5

Надежда ниже будет работать:

@a.map!{|n| n == i ? (@b.push n; 2*n) : n }