Module: DatasetsPandas::DatasetConversion

Included in:
Datasets::Dataset
Defined in:
lib/datasets-pandas/conversion.rb

Instance Method Summary collapse

Instance Method Details

#to_pandas(*keys) ⇒ Object



5
6
7
8
9
10
11
12
# File 'lib/datasets-pandas/conversion.rb', line 5

def to_pandas(*keys)
  if keys.empty?
    Pandas::DataFrame.new(data: self.map {|record| record.to_h })
  else
    records = self.map {|record| keys.map {|k| record[k] } }
    Pandas::DataFrame.new(data: records, columns: keys)
  end
end