Module: DatasetsNumoNArray::DatasetNArrayable

Included in:
Datasets::Dataset
Defined in:
lib/datasets-numo-narray/narrayable.rb

Instance Method Summary collapse

Instance Method Details

#to_narray(*keys) ⇒ Object



5
6
7
8
9
10
11
12
13
14
15
16
17
18
# File 'lib/datasets-numo-narray/narrayable.rb', line 5

def to_narray(*keys)
  if keys.empty?
    array = collect do |record|
      record.values
    end
  else
    array = collect do |record|
      keys.collect do |key|
        record[key]
      end
    end
  end
  Numo::NArray[*array]
end