2011-04-16 106 views
17

根据某些条件将数组拆分为多个数组的最简单方法是什么?在我的场景中,我需要将整数和字符串值移动到不同的数组。我试过split方法,但不能按预期工作。在ruby中将数组拆分为多个小数组的最佳方法

x=[1,2,3,"a","b",4] 
x.split {|item| item.kind_of? Fixnum} 

在C#中,存在一个Linq选择group by,它可以帮助您组基于条件的对象。 Object(不使用activerecord)有没有类似的方法?

有没有简单的方法?

回答

38

您正在寻找Enumerable#partition

x = [1, 2, 3, "a", "b", 4] 
numbers, not_numbers = x.partition{|item| item.kind_of?(Fixnum)} 
# => [[1, 2, 3, 4], ["a", "b"]] 
+0

+ 1为单线和双重任务。看起来不错。 – slhck 2011-04-16 12:45:47

+0

+1这对于速度来说也是一个可靠的胜利!查看我添加的基准。 – 2011-04-17 04:16:47

+3

下面的答案使用'group_by',它允许分成两个以上的组。 – 2013-10-26 04:02:05

3

这里是我的解决方案:

hash = x.group_by { |t| t.kind_of? Fixnum } 
# hash => {true=>[1, 2, 3, 4], false=>["a", "b"]} 
array1 = hash[true] # The array of integers 
array2 = hash[false] # The array of strings 
5

尝试:

x.group_by {|x| x.class} 

您可以通过然后调用to_a得到一个数组回来在你给出的例子中的结果将返回:

[[Fixnum, [1, 2, 3, 4]], [String, ["a", "b"]]] 
5

只是一些更多的解决方案扔到池:

x = [1,2,3,"a","b",4] 

numbers = x.select{ |e| e.is_a?(Fixnum) } # => [1, 2, 3, 4] 
letters = x - numbers # => ["a", "b"] 

numbers = x.select{ |e| e.kind_of?(Fixnum) } # => [1, 2, 3, 4] 
letters = x - numbers # => ["a", "b"] 

(numbers, letters) = x.group_by {|a| a.class}.values_at(Fixnum, String) 
numbers # => [1, 2, 3, 4] 
letters # => ["a", "b"] 

随着一些基准测试显示如何了微妙的变化影响速度:

require 'benchmark' 

x = [1,2,3,"a","b",4] * 100 
n = 10_000 
Benchmark.bm do |bench| 
    bench.report { n.times { 
    numbers = x.select{ |e| e.is_a?(Fixnum) } 
    letters = x - numbers 
    }} 

    bench.report { n.times { 
    numbers = x.select{ |e| e.kind_of?(Fixnum) } 
    letters = x - numbers 
    }} 

    bench.report { n.times { 
    (numbers, letters) = x.group_by {|a| a.class}.values_at(Fixnum, String) 
    }} 

    bench.report { n.times { 
    numbers, not_numbers = x.partition{|item| item.kind_of? Fixnum} 
    }} 
end 
# >>  user  system  total  real 
# >> 4.270000 0.010000 4.280000 ( 4.282922) 
# >> 4.290000 0.000000 4.290000 ( 4.288720) 
# >> 5.160000 0.010000 5.170000 ( 5.163695) 
# >> 3.720000 0.000000 3.720000 ( 3.721459) 
+0

奇怪的是,我发现'.partition'更快,甚至创建2个数组,并且执行'.each'和'if e.kind_of(Fixnum)'。我猜他们正在为C中的'.partition'做一些优化。 – Dogbert 2011-04-17 09:35:35

+0

我没有看过源代码,但我有同样的怀疑。 – 2011-04-17 09:47:56

相关问题