Làm cách nào để kết xuất mảng 2D trực tiếp vào tệp CSV?

Question

Tôi có mảng 2D này:

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

tôi thường làm:

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

Có cách nào dễ dàng hơn hoặc trực tiếp làm việc đó khác vì thêm từng hàng?

Answer 1

Giả sử rằng mảng của bạn chỉ là con số (không dây có khả năng có dấu phẩy trong họ) thì:

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

Một chuỗi khổng lồ blatted vào đĩa, không có sự tham gia của các thư viện CSV.

Ngoài ra, sử dụng thư viện CSV để correctly escape each row:

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 } 

Nếu bạn phải làm điều này thường xuyên và mã phiền bạn, bạn có thể monkeypatch nó trong:

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

Mở rộng câu trả lời trên bởi @Phrogz, trong khi sử dụng thư viện csv và yêu cầu thay đổi dấu phân cách mặc định:

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