ruby-on-rails - 如何使用 Ruby 替换 CSV 中指定列的值？

Question

我有一个 CSV，其中第一列给出了工作日。但问题是我需要在 MATLAB 中使用 CSV，它不允许字符串值。所以，我想在 CSV 中将星期日转换为 7，...，星期一转换为 1。

但我似乎无法在 Ruby 中找到一种方法来做到这一点。而且我也无法手动打开 excel 来执行此操作，因为文件很大。具体来说，我在弄清楚如何在 Ruby 中访问和指向我想要的特定列的语法时遇到了问题。例如，如果文件已加载到 MATLAB 中，并且出于某种原因仍需要我将工作日转换为数字，我会编写如下简单代码:

for i=1:length(Columns(:,1))
    if Columns(i,1)=='sunday'
        Columns(i,1)=7
    elseif Columns(i,1)=='saturday'
        Columns(i,1)=6
    elseif Columns(i,1)=='friday'
        Columns(i,1)=5
    elseif Columns(i,1)=='thursday'
        Columns(i,1)=4
    elseif Columns(i,1)=='wednesday'
        Columns(i,1)=3
    elseif Columns(i,1)=='tuesday'
        Columns(i,1)=2
    elseif Columns(i,1)=='monday'
        Columns(i,1)=1
    end
end

所以，我在找出与此语句等效的 Ruby 时遇到了问题:

for i=1:length(Columns(:,1))

感谢任何帮助。谢谢。

Answer 1

Ruby 中有两个很好的 CSV 库，根据您所说的，我假设 CSV 文件没有 header (如果有，SmarterCSV 会使事情变得更容易一些)。

也就是说，您想要获取第一列:

require 'csv'

your_csv = CSV.open("your_csv.csv")
# This is the line you wanted:
first_column = your_csv.map(&:first)

# Then to do the weekday conversion (with a Hash):
convert_weekdays = { "sunday" => 1, "monday" => 2, "tuesday" => 3, "wednesday" => 4, "thursday" => 5, "friday" => 6, "saturday" => 7 }

converted = first_column.map { |row| convert_weekdays[row] }

不确定这是否正是您想要的，在 Ruby 中有很多方法可以处理 CSV 文件。

要保存 CSV，您需要使用 CSV.open 打开一个新的(或相同的)CSV 文件:

your_csv = CSV.open("your_csv.csv")
CSV.open("saved_csv.csv", "w") do |csv|
  your_csv.each { |row| csv << [convert_weekdays[row.first], *row[1...row.size]] }
end

抱歉有点不雅，用 Ruby 编写 CSV 并不总是最简单的事情!请注意花括号 {} 与 do end 相同，但通常在内部 block 只有一行时使用。

编辑:对于大文件，此方法可能会更快一些:

your_csv = CSV.parse("your_csv.csv")
convert_weekdays = { "sunday" => 1, "monday" => 2, "tuesday" => 3, "wednesday" => 4, "thursday" => 5, "friday" => 6, "saturday" => 7 }
by_columns = your_csv.transpose
by_columns.first.map! { |row| convert_weekdays[row] }
CSV.open("saved_csv.csv", "w") do |csv|
  by_columns.transpose.each { |row| csv << row }
end

这样，您将 CSV 作为二维数组加载，转置它，并且只对第一列进行操作。