Class: Charty::PandasIndex

Inherits:

Index

• Object
• Index
• Charty::PandasIndex

Defined in:

lib/charty/index.rb

Instance Attribute Summary

Attributes inherited from Index

#name, #values

Class Method Summary

• .try_convert(obj) ⇒ Object

Instance Method Summary

• #==(other) ⇒ Object
• #each(&block) ⇒ Object
• #length ⇒ Object
• #loc(key) ⇒ Object
• #union(other) ⇒ Object

Methods inherited from Index

#[], #initialize

Constructor Details

This class inherits a constructor from Charty::Index

Class Method Details

.try_convert(obj) ⇒ Object

# File 'lib/charty/index.rb', line 145

def self.try_convert(obj)
  case obj
  when PandasIndex
    obj
  when ->(x) { defined?(Pandas) && x.is_a?(Pandas::Index) }
    PandasIndex.new(obj)
  when RangeIndex, Range
    obj = obj.values if obj.is_a?(RangeIndex)
    stop = if obj.exclude_end?
             obj.end
           else
             obj.end + 1
           end
    PandasIndex.new(Pandas.RangeIndex.new(obj.begin, stop))
  when ->(x) { defined?(Enumerator::ArithmeticSequence) && x.is_a?(Enumerator::ArithmeticSequence) }
    stop = if obj.exclude_end?
             obj.end
           else
             obj.end + 1
           end
    PandasIndex.new(Pandas::RangeIndex.new(obj.begin, stop, obj.step))
  when Index, Array, DaruIndex, ->(x) { defined?(Daru) && x.is_a?(Daru::Index) }
    obj = obj.values if obj.is_a?(Index)
    PandasIndex.new(Pandas::Index.new(obj.to_a))
  else
    nil
  end
end

Instance Method Details

#==(other) ⇒ Object

# File 'lib/charty/index.rb', line 180

def ==(other)
  case other
  when PandasIndex
    Numpy.all(values == other.values)
  when Index
    return false if length != other.length
    Numpy.all(values == other.values.to_a)
  else
    super
  end
end

#each(&block) ⇒ Object

# File 'lib/charty/index.rb', line 192

def each(&block)
  return enum_for(__method__) unless block_given?

  i, n = 0, length
  while i < n
    yield self[i]
    i += 1
  end
end

#length ⇒ Object

# File 'lib/charty/index.rb', line 176

def length
  size
end

#loc(key) ⇒ Object

# File 'lib/charty/index.rb', line 202

def loc(key)
  case values
  when Pandas::Index
    values.get_loc(key)
  else
    super
  end
end

#union(other) ⇒ Object

# File 'lib/charty/index.rb', line 211

def union(other)
  other = PandasIndex.try_convert(other)
  # NOTE: Using `sort=False` in pandas.Index#union does not produce pandas.RangeIndex.
  # TODO: Reconsider to use `sort=True` here.
  PandasIndex.new(values.union(other.values, sort: false))
end