我有两个清单:
a = [1,2,3]
b = ["a","b","c"]
我的列表l
是:
l = [a,b].flatten
所以l = [1,2,3,"a","b","c"]
我正在寻找一种按照其中的项目类型拆分列表的优雅方式,以便a
和b
恢复原状。
我可以使用列表中的每个项目进行测试,但这似乎不是在运行时方面有效,也不是在代码方面。
答案 0 :(得分:6)
您可以使用group_by
,然后从生成的哈希中拉出阵列:
>> by_class = l.group_by(&:class)
=> {Integer=>[1, 2, 3], String=>["a", "b", "c"]}
>> a = by_class[Fixnum]
=> [1, 2, 3]
>> b = by_class[String]
=> ["a", "b", "c"]
如果您知道自己只有Fixnums和字符串,则可以使用partition
:
>> a, b = *l.partition { |o| o.is_a? Fixnum }
=> [[1, 2, 3], ["a", "b", "c"]]
>> a
=> [1, 2, 3]
>> b
=> ["a", "b", "c"]