Array與模運算符不起作用

Question

第一個代碼有效，但我不明白爲什麼第二個代碼沒有。任何洞察力將不勝感激。我知道在這個例子中我真的不需要一個數組，我只是想爲了學習而努力。

def stamps(input) 
    if input % 5 == 0 
    puts 'Zero!' 
    else 
    puts 'NO!' 
    end 
end 

print stamps(8) 

但這不工作：

array_of_numbers = [8] 

def stamps(input_array) 
    if input_array % 5 == 0 
    puts 'Zero!' 
    else 
    puts 'NO!' 
    end 
end 

print stamps(array_of_numbers) 

Answer 1

因爲input_array是一個數組，圖8是一個數字。使用first來檢索數組的第一個元素。

array_of_numbers = [8] 

def stamps(input_array) 
    if input_array.first % 5 == 0 
    puts 'Zero!' 
    else 
    puts 'NO!' 
    end 
end 

print stamps(array_of_numbers) 

Answer 2

下面的函數的情況下，工作原理是輸入數字或數組：

def stamps(input) 
    input = [input] unless input.is_a?(Array) 
    if input.first % 5 == 0 
    puts 'Zero!' 
    else 
    puts 'NO!' 
    end 
end