我該如何重構這個ruby代碼呢？

Question

我在試圖反應這段代碼。有什麼建議？

@contest_users = [] 
@contest.squads.each do |squad| 
    @contest_users << squad.users[0] 
end 

Answer 1

使用map而不是each：

@contest_users = @contest.squads.map { |squad| squad.users.first } 

Answer 2

@contest_users = @contest.squads.each_with_object([]) do |squad, array| 
    array << squad.users[0] 
end