Class: Daru::DataFrame

Inherits:
Object
  • Object
show all
Defined in:
lib/to_csv.rb

Instance Method Summary collapse

Instance Method Details

#add_vector(vecname, vec) ⇒ Object Also known as: addvec

To avoid bug about adding column to Daru::DataFrame



23
24
25
26
 
# File 'lib/to_csv.rb', line 23

def add_vector(vecname, vec)
  self[vecname] = vec
  self.rename_vectors({vecname => vecname})
end

#convert_enc!(from: "cp932", to: "utf-8") ⇒ Object

ver.0.3.8~ Convert Daru::DF encoding



42
43
44
45
46
47
48
 
# File 'lib/to_csv.rb', line 42

def convert_enc!(from: "cp932", to: "utf-8")
  self.vectors.each do |col|
    self[col] = self[col].each {|val| val.encode!(to, from_encoding: from) if val.is_a?(String)}
  end
  
  self.encode_vectors!
end

#encode_vector_name(vector_i) ⇒ Object

エンコード関連 ### vector_i番目のヘッダー名を読めるようにエンコード



30
31
32
33
34
 
# File 'lib/to_csv.rb', line 30

def encode_vector_name(vector_i)
  if self.vectors.to_a[vector_i].is_a?(String)
    self.vectors.to_a[vector_i].encode Encoding::UTF_8, Encoding::Windows_31J
  end
end

#encode_vectors! ⇒ Object

すべての列に対し上記を実施



37
38
39
 
# File 'lib/to_csv.rb', line 37

def encode_vectors!
  self.vectors = Daru::Index.new(Range.new(0, self.vectors.size-1).map {|i| encode_vector_name i })
end

#set_index!(indexcolumn) ⇒ Object

rover not suppoted yet about indexing



52
53
54
 
# File 'lib/to_csv.rb', line 52

def set_index!(indexcolumn)
  self.index = self[indexcolumn]
end

#simplify_multi_index(vector_names_ary) ⇒ Object

To revice pivot index



57
58
59
60
 
# File 'lib/to_csv.rb', line 57

def simplify_multi_index(vector_names_ary)
  self.vectors = Daru::Index.new(vector_names_ary)
  self.index = Daru::Vector.new(self.index.to_a.map{_1[0]})
end

#to_csv ⇒ Object 



5
6
7
8
9
10
11
12
13
14
15
 
# File 'lib/to_csv.rb', line 5

def to_csv()
  a = self.to_a.transpose
  
  ans = self.map(&:name).join ","
  self.to_a[0].each do |item|
    ans += "\n"
    ans += item.map{|k, v| v}.join(",")
  end
  
  return ans
end

#write_csv(path, encoding: nil) ⇒ Object 



17
18
19
20
 
# File 'lib/to_csv.rb', line 17

def write_csv(path, encoding: nil)
  enc = encoding.nil? ? "" : ":#{encoding}"
  open(path, "w#{enc}") { _1.write to_csv }
end