紅寶石地圖！功能和嵌套陣列

Question

我需要找到每個嵌套元件的所述第二和第三元件之間的距離，以便

nested_array = [[0, 3, 4], [1, 20, 21], [2, 2, 2]] 

def pythag_theorem(a, b) 
    c = (a * a) + (b * b) 
    result = Math.sqrt(c) 
    result 
end 

def find_distance(array) 
    t = 0 
    while t < array.length 
    array[t].map! {|x| pythag_theorem(x[1], x[2])} 
    t += 1 
    end 
array 
end 

print find_distance(nested_array) 

我得到

[[0.0, 1.4142135623730951, 0.0], [1.0, 0.0, 1.0], [1.0, 1.0, 1.0]] 

當我需要

[[0, 5], [1, 29], [2, 2.82842712474619]] 

pythag_theorem工作，但爲什麼不是地圖！爲我工作？謝謝。

Answer 1

a = [[0, 3, 4], [1, 20, 21], [2, 2, 2]] 
a.map {|x,y,z| [x, Math.sqrt(y*y + z*z)]} 
# => [[0, 5.0], [1, 29.0], [2, 2.82842712474619]] 

Answer 2

地圖爲你做循環。您不需要while循環。您也可能不應使用map!修改陣列。

array = [[0, 3, 4], [1, 20, 21], [2, 2, 2]] 
array.map { |(_, a, b)| Math.sqrt(a * a + b * b) } 
# => [5.0, 29.0, 2.8284271247461903] 

Answer 3

這應該爲你工作：

def find_distance(array) 
    array.map do |tuple| 
    [tuple[0],pythag_theorem(tuple[1],tuple[2])] 
    end 
end 

BTW是不修改到位函數的參數是一個好主意。