Sha256: f9ea2b683cf2a4ff88f25ea249079ade1e35cda19db89641971b863725346a35

Contents?: true

Size: 1.45 KB

Versions: 26

Compression:

Stored size: 1.45 KB

Contents

module Enumerable

  # Returns an array of elements for the elements that occur n times.
  # Or according to the results of a given block.
  #
  #   [1,1,2,3,3,4,5,5].occur(1)             #=> [2,4]
  #   [1,1,2,3,3,4,5,5].occur(2)             #=> [1,3,5]
  #   [1,1,2,3,3,4,5,5].occur(3)             #=> []
  #
  #   [1,2,2,3,3,3].occur(1..1)              #=> [1]
  #   [1,2,2,3,3,3].occur(2..3)              #=> [2,3]
  #
  #   [1,1,2,3,3,4,5,5].occur { |n| n == 1 } #=> [2,4]
  #   [1,1,2,3,3,4,5,5].occur { |n| n > 1 }  #=> [1,3,5]
  #
  def occur(n=nil) #:yield:
    result = Hash.new { |hash, key| hash[key] = Array.new }
    self.each do |item|
      key = item
      result[key] << item
    end
    if block_given?
      result.reject! { |key, values| ! yield(values.size) }
    else
      raise ArgumentError unless n
      if Range === n
        result.reject! { |key, values| ! n.include?(values.size) }
      else
        result.reject! { |key, values| values.size != n }
      end
    end
    return result.values.flatten.uniq
  end

end



#  _____         _
# |_   _|__  ___| |_
#   | |/ _ \/ __| __|
#   | |  __/\__ \ |_
#   |_|\___||___/\__|
#
=begin test

  require 'test/unit'

  class TCEnumerable < Test::Unit::TestCase

    def test_occur
      arr = [:a,:b,:a]
      assert_equal( [:b], arr.occur(1) )
      assert_equal( [:a], arr.occur(2) )
      assert_equal( [:b], arr.occur(1..1) )
      assert_equal( [:a], arr.occur{ |n| n % 2 == 0 } )
    end

  end

=end

Version data entries

26 entries across 26 versions & 1 rubygems

Version Path
facets-1.0.3 packages/core/lib/facet/enumerable/occur.rb
facets-1.0.0 lib/facet/enumerable/occur.rb
facets-0.9.0 lib/nano/enumerable/occur.rb
facets-1.2.0 lib/facets/core/enumerable/occur.rb
facets-1.2.1 lib/facets/core/enumerable/occur.rb
facets-1.1.0 lib/facet/enumerable/occur.rb
facets-1.3.0 lib/facets/core/enumerable/occur.rb
facets-1.3.1 lib/facets/core/enumerable/occur.rb
facets-1.3.3 lib/facets/core/enumerable/occur.rb
facets-1.3.2 lib/facets/core/enumerable/occur.rb
facets-1.4.1 lib/facets/core/enumerable/occur.rb
facets-1.4.0 lib/facets/core/enumerable/occur.rb
facets-1.4.2 lib/facets/core/enumerable/occur.rb
facets-1.4.3 lib/facets/core/enumerable/occur.rb
facets-1.4.4 lib/facets/core/enumerable/occur.rb
facets-1.4.5 lib/facets/core/enumerable/occur.rb
facets-1.7.38 lib/facets/core/enumerable/occur.rb
facets-1.7.0 lib/facets/core/enumerable/occur.rb
facets-1.7.30 lib/facets/core/enumerable/occur.rb
facets-1.7.46 lib/facets/core/enumerable/occur.rb