在Ruby中从另一个数组中减去一个数组

时间:2009-07-28 06:14:02

标签: ruby arrays array-difference

我有两个任务数组 - 创建和分配。 我想从创建的任务数组中删除所有已分配的任务。 这是我的工作,但很乱,代码:

    @assigned_tasks = @user.assigned_tasks
    @created_tasks = @user.created_tasks

    #Do not show created tasks assigned to self
    @created_not_doing_tasks = Array.new
    @created_tasks.each do |task|
        unless @assigned_tasks.include?(task)
            @created_not_doing_tasks << task
        end
    end

我确信有更好的方法。它是什么? 谢谢: - )

2 个答案:

答案 0 :(得分:158)

你可以在Ruby中减去数组:

[1,2,3,4,5] - [1,3,4]  #=> [2,5]
  

ary - other_ary→new_ary数组差异

     

返回一个新数组,它是原始数组的副本,删除任何数组   也出现在other_ary中的项目。订单保留在   原始阵列。

     

它使用hash和eql来比较元素?提高效率的方法。

     

[ 1, 1, 2, 2, 3, 3, 4, 5 ] - [ 1, 2, 4 ] #=> [ 3, 3, 5 ]

     

如果您需要   类似集合的行为,请参阅库类Set。

请参阅Array文档。

答案 1 :(得分:5)

上述解决方案

a - b

从数组b删除数组a中元素的所有实例。

[ 1, 1, 2, 2, 3, 3, 4, 5 ] - [ 1, 2, 4 ]  #=>  [ 3, 3, 5 ]

在某些情况下,您希望结果为[1, 2, 3, 3, 5]。也就是说,您不希望删除所有重复项,而只删除元素。

您可以通过

实现这一目标
class Array
  def delete_elements_in(ary)
    ary.each do |x|
      if index = index(x)
        delete_at(index)
      end
    end
  end
end

测试

irb(main):198:0> a = [ 1, 1, 2, 2, 3, 3, 4, 5 ]
=> [1, 1, 2, 2, 3, 3, 4, 5]
irb(main):199:0> b = [ 1, 2, 4 ]
=> [1, 2, 4]
irb(main):200:0> a.delete_elements_in(b)
=> [1, 2, 4]
irb(main):201:0> a
=> [1, 2, 3, 3, 5]

即使两个数组未排序,代码也能正常工作。在示例中,数组已排序,但这不是必需的。