ruby - 如何将二维数组直接转储到 CSV 文件中？

Question

我有这个二维数组：

arr = [[1,2],[3,4]]

我通常这样做：

CSV.open(file) do |csv| 
  arr.each do |row| 
    csv << row
  end
end

除了逐行添加之外，还有其他更简单或直接的方法吗？

Answer 1

假设您的数组只是数字（没有可能包含逗号的字符串），那么：

File.open(file,'w'){ |f| f << arr.map{ |row| row.join(',') }.join('\n') }

一个巨大的字符串被写入磁盘，不涉及 CSV 库。

或者，使用 CSV 库正确转义每一行：

require 'csv'
# #to_csv automatically appends '\n', so we don't need it in #join
File.open(file,'w'){ |f| f << arr.map(&:to_csv).join } 

如果您必须经常这样做并且代码困扰您，您可以将其修改为：

class CSV
  def self.dump_array(array,path,mode="rb",opts={})
    open(path,mode,opts){ |csv| array.each{ |row| csv << row } }
  end
end
CSV.dump_array(arr,file)

Answer 2

通过@Phrogz 扩展上面的答案，同时使用 csv 库并需要更改默认分隔符：

File.open(file,'w'){ |f| f << arr.map{|x| x.to_csv(col_sep: '|')}.join }