我需要将 3 列从一个巨大的 csv 文件复制到一个新的 csv 文件中。怎么做？

Question

我对红宝石很陌生。这是我的代码：

require 'csv'

csv = CSV.read('C:\Users\hp1\Desktop\Datasets\000.csv', :headers=>true)

p csv['timestamp']
f = File.new('C:\Users\hp1\Desktop\Datasets\Columns_000.csv', 'wb')
    f.puts(p)
    f.close

我想要在这段代码中做的是，我想读取 000.csv 文件，并将名为“timestamp”的列复制到另一个 csv，Columns_000.csv。

但是发生的情况是，它会在我的终端上写入时间戳列的值。相反，我希望它把它们写在新的 csv 上。

我还有 2 列像我想复制到新的 csv 一样。有没有办法将它们全部复制在一起，还是我需要一个一个地复制它们？

Answer 1

COLUMNS = ['timestamp', 'other1', 'other2']

# open new csv for writing
CSV.open("C:\Users\hp1\Desktop\Datasets\000_out.csv", "wb") do |csv|
  # iterating existing csv rows
  CSV.foreach('C:\Users\hp1\Desktop\Datasets\000.csv', :headers=>true) do |row|
    # select only those specified above columns
    csv << COLUMNS.map { |col| row[col] }
  end
end