如何重构此 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