Выходной массив в CSV в Ruby

Достаточно легко прочитать CSV файл в массиве с Ruby, но я не могу найти хорошую документацию о том, как писать массив в CSV файл. Может ли кто-нибудь сказать мне, как это сделать?

Я использую Ruby 1.9.2, если это имеет значение.

Ответ 1

В файл:

require 'csv'
CSV.open("myfile.csv", "w") do |csv|
  csv << ["row", "of", "CSV", "data"]
  csv << ["another", "row"]
  # ...
end

В строку:

require 'csv'
csv_string = CSV.generate do |csv|
  csv << ["row", "of", "CSV", "data"]
  csv << ["another", "row"]
  # ...
end

Здесь текущая документация по CSV: http://ruby-doc.org/stdlib/libdoc/csv/rdoc/index.html

Ответ 2

У меня это до одной строки.

rows = [['a1', 'a2', 'a3'],['b1', 'b2', 'b3', 'b4'], ['c1', 'c2', 'c3'], ... ]
csv_str = rows.inject([]) { |csv, row|  csv << CSV.generate_line(row) }.join("")
#=> "a1,a2,a3\nb1,b2,b3\nc1,c2,c3\n" 

Сделайте все вышеперечисленные и сохраненные в csv в одной строке.

File.open("ss.csv", "w") {|f| f.write(rows.inject([]) { |csv, row|  csv << CSV.generate_line(row) }.join(""))}

Примечание:

Для преобразования активной базы данных записей в csv было бы что-то вроде этого, я думаю

CSV.open(fn, 'w') do |csv|
  csv << Model.column_names
  Model.where(query).each do |m|
    csv << m.attributes.values
  end
end

Hmm @tamouse, этот gist несколько запутывает меня, не читая исходный код csv, но в целом, если каждый хэш в вашем массиве имеет одинаковое количество пар k/v и что ключи всегда одинаковы, в том же (т.е. если ваши данные структурированы), это должно сделать:

rowid = 0
CSV.open(fn, 'w') do |csv|
  hsh_ary.each do |hsh|
    rowid += 1
    if rowid == 1
      csv << hsh.keys# adding header row (column labels)
    else
      csv << hsh.values
    end# of if/else inside hsh
  end# of hsh (rows)
end# of csv open

Если ваши данные не структурированы, это явно не сработает

Ответ 3

Если кому-то интересно, вот несколько однострочных (и примечание о потере информации о типе в CSV):

require 'csv'

rows = [[1,2,3],[4,5]]                    # [[1, 2, 3], [4, 5]]

# To CSV string
csv = rows.map(&:to_csv).join             # "1,2,3\n4,5\n"

# ... and back, as String[][]
rows2 = csv.split("\n").map(&:parse_csv)  # [["1", "2", "3"], ["4", "5"]]

# File I/O:
filename = '/tmp/vsc.csv'

# Save to file -- answer to your question
IO.write(filename, rows.map(&:to_csv).join)

# Read from file
# rows3 = IO.read(filename).split("\n").map(&:parse_csv)
rows3 = CSV.read(filename)

rows3 == rows2   # true
rows3 == rows    # false

Примечание: CSV теряет всю информацию о типе, вы можете использовать JSON для сохранения информации о базовом типе или перейти к подробному (но более легко редактируемому человеку) YAML, чтобы сохранить всю информацию о типе - например, если вам нужен тип даты, который станет строками в CSV и JSON.

Ответ 4

Если у вас есть массив массивов данных:

rows = [["a1", "a2", "a3"],["b1", "b2", "b3", "b4"], ["c1", "c2", "c3"]]

Затем вы можете записать это в файл со следующим, что, я думаю, намного проще:

require "csv"
File.write("ss.csv", rows.map(&:to_csv).join)

Ответ 5

Основываясь на ответе @boulder_ruby, это то, что я ищу, предполагая, что us_eco содержит таблицу CSV как из моего сущности.

CSV.open('outfile.txt','wb', col_sep: "\t") do |csvfile|
  csvfile << us_eco.first.keys
  us_eco.each do |row|
    csvfile << row.values
  end
end

Обновлен метод https://gist.github.com/tamouse/4647196

Ответ 6

Борясь с этим сам. Это мое взятие:

https://gist.github.com/2639448:

require 'csv'

class CSV
  def CSV.unparse array
    CSV.generate do |csv|
      array.each { |i| csv << i }
    end
  end
end

CSV.unparse [ %w(your array), %w(goes here) ]