我该如何重构这个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