如何循环条件以将新值分配给其他属性?

时间:2013-03-04 21:42:01

标签: ruby loops where

我正在尝试使用:name属性中的文本在新的哈希数组中指定unit_type。

这是我的数据

class Unit
  attr_accessor :name
  attr_accessor :imported_id
  attr_accessor :country
  attr_accessor :unit_type

  raw_stuff = [{:old_id=>576, :name=>"16th Armored Division (USA) "}, {:old_id=>578, :name=>"20th Armored Division (USA)"}, {:old_id=>759, :name=>"27th Armoured Brigade (UK)"}, {:old_id=>760, :name=>"- 13th/18th Royal Hussars"}, {:old_id=>761, :name=>"- East Riding of Yorkshire Yeomanry "}, {:old_id=>762, :name=>"- Staffordshire Yeomanry "}, {:old_id=>769, :name=>"A I R B O R N E "}, {:old_id=>594, :name=>"1st Airborne Division (UK)"}, {:old_id=>421, :name=>"6th Airborne Division (UK)"}]

  units = []

  raw_stuff.each do |unit_hash|
   u = Unit.new
   u.name = unit_hash[:name].sub("-","").lstrip
   u.unit_type = unit_hash[:name].scan("Division")
   puts u.unit_type
   puts u.name
  end

end

这适当地将“division”指定为unit_type。 但是,我似乎无法分配任何其他内容,例如“旅”。 我应该使用if或where条件吗?

When I use the following code:
  raw_stuff.each do |unit_hash|
   u = Unit.new
   u.name = unit_hash[:name].sub("-","").lstrip
      if unit_hash[:name].scan("Division")
        u.unit_type = "Division"
      elsif unit_hash[:name].scan("Brigade")
        u.unit_hash = "Brigade"
      else
        u.unit_hash = nil
      end
   puts u.unit_type
   puts u.name
  end

我最终将Divison分配到每个单位。

2 个答案:

答案 0 :(得分:1)

可爱的单行:

u.unit_type = unit_hash[:name][/Division|Brigade/]

代码中的错误是scan在找不到任何内容时返回一个空数组([]),而一个空数组是“真实的”。您正在寻找的方法是include?我的解决方案通过直接将字符串搜索结果(可以是nil)分配给单元类型来完全绕过条件。

答案 1 :(得分:0)

试试这个:

if unit_hash[:name].include?("Division")
    u.unit_type = "Division"
  elsif unit_hash[:name].include?("Brigade")
    u.unit_type = "Brigade"
  else
    u.unit_type = nil
  end