Contents

# lib/sqa/data_frame.rb
# frozen_string_literal: true

require_relative  'data_frame/yahoo_finance'

class Daru::DataFrame

  def to_csv(path_to_file, opts={})
    options = {
      converters: :numeric
    }.merge(opts)

    writer = ::CSV.open(path_to_file, 'wb')

    writer << vectors.to_a unless options[:headers] == false

    each_row do |row|
      writer << if options[:convert_comma]
                  row.map { |v| v.to_s.tr('.', ',') }
                else
                  row.to_a
                end
    end

    writer.close
  end
end




class SQA::DataFrame < Daru::DataFrame


  #################################################

  def self.path(filename)
    Pathname.new(SQA.config.data_dir) + filename
  end

  def self.load(filename, options={}, &block)
    source  = path(filename)
    type    = source.extname.downcase

    if ".csv" == type
     from_csv(source, options={}, &block)
    elsif ".json" == type
      from_json(source, options={}, &block)
    elsif %w[.txt .dat].include?(type)
      from_plaintext(source, options={}, &block)
    elsif ".xls" == type
      from_excel(source, options={}, &block)
    else
      raise SQA::BadParamenterError, "un-suppod file type: #{type}"
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
sqa-0.0.12	lib/sqa/data_frame.rb