如何使用 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 文件没有标题（如果有，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 相同，但通常在内部块只有一行时使用。

编辑：这种方法对于大文件可能会快一点：

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 加载为二维数组，对其进行转置，然后仅对第一列进行操作。