我正在尝试使用: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分配到每个单位。
答案 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