在ruby中按类或kind_of分隔列表项

Question

我有两个清单:

a = [1,2,3]
b = ["a","b","c"]

我的清单l是:

l = [a,b].flatten 

所以 l = [1,2,3,"a","b","c"]

我正在寻找在它的项目类型划分的列表中,以便有一个优雅的方式a,并b恢复原样.

我可以使用列表中的每个项目并进行测试,但这似乎不是运行时有效也不是代码.

Answer 1

你可以使用group_by然后从生成的Hash中拉出你的数组:

>> 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和Strings那么你可以使用partition:

>> a, b = *l.partition { |o| o.is_a? Fixnum }
=> [[1, 2, 3], ["a", "b", "c"]]
>> a
=> [1, 2, 3]
>> b
=> ["a", "b", "c"]