在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，然后从生成的哈希中拉出阵列：

>> 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"]