Ruby on Rails人口统计数据

时间:2011-09-16 09:01:27

标签: ruby-on-rails ruby statistics

我为PS3游戏制作了一个网站,我有很多用户。我想根据人们的位置制作比赛,并且还希望针对年龄组。当用户以YYYY-MM-DD格式注册输入时的出生日期。我正在拉取数据并将其变成像这样的哈希:

# Site.rb

has_many :members

def ages
  ages = {"Under 18" => 0, "19-24" => 0, "25-35" => 0, "36-50" => 0, "51-69" => 0,"70+" => 0}
  ages_results = self.members.count("DATE_FORMAT(dob, '%Y')", :group =>"DATE_FORMAT(dob, '%Y')")
  ages_results.each do |k,v|
    k = k.to_i
    if k.between?(18.years.ago.strftime("%Y").to_i, 0.years.ago.strftime("%Y").to_i)
      ages["Under 18"] += v
    elsif k.between?(24.years.ago.strftime("%Y").to_i, 19.years.ago.strftime("%Y").to_i)
      ages["19-24"] += v
    elsif k.between?(35.years.ago.strftime("%Y").to_i, 25.years.ago.strftime("%Y").to_i)
      ages["25-35"] += v
    elsif k.between?(50.years.ago.strftime("%Y").to_i, 36.years.ago.strftime("%Y").to_i)
      ages["36-50"] += v
    elsif k.between?(69.years.ago.strftime("%Y").to_i, 51.years.ago.strftime("%Y").to_i)
      ages["51-69"] += v 
    elsif k > 70.years.ago.strftime("%Y").to_i
      ages["70+"] += v
    end
  end
  ages
end

我不是专家ruby开发人员,不确定上述方法是否合适,或者可以采用更好的方式,有人可以给我一些建议吗?

干杯

1 个答案:

答案 0 :(得分:3)

您的代码中需要注意的事项:

  • 你似乎无视用户出生的月和日
  • 您不必要地转换为字符串和从字符串转换:

    50.years.ago.strftime("%Y").to_i
    

    可以写成

    50.years.ago.year
    
  • 遍及代码的硬编码值

我会通过找到一个计算确切年龄的适当方法来开始重写。 This one似乎没问题:

require 'date'

def age(dob)
  now = Time.now.utc.to_date
  now.year - dob.year - ((now.month > dob.month || (now.month == dob.month && now.day >= dob.day)) ? 0 : 1)
end

然后我会将年龄表提取到一个单独的结构中,以便能够在需要时轻松更改它,并将它一起视觉化:

INF = 1/0.0 # convenient infinity
age_groups = {
  (0..18) => 'Under 18',
  (19..24) => '19-24',
  (25..35) => '25-35',
  (36..50) => '36-50',
  (51..69) => '51-69',
  (70..INF) => '70+'
}

接下来,您可以将用户出生日期数组作为输入:

users_dobs = [Date.new(1978,4,16), Date.new(2001,6,13), Date.new(1980,10,22)]

并开始根据您的地图找到合适的方法对其进行分组,例如使用inject

p users_dobs.each_with_object({}) {|dob, result|
  age_group = age_groups.keys.find{|ag| ag === age(dob)}
  result[age_group] ||= 0
  result[age_group] += 1
}
#=>{25..35=>2, 0..18=>1}

或者,或许,使用group_by

p users_dobs.group_by{|dob|
  age_groups.keys.find{|ag| ag === age(dob)}
}.map{|k,v| [age_groups[k], v.count]}
#=>[["25-35", 2], ["Under 18", 1]]