如何为CSV文件添加'Total'列并正确处理表头？

嘿，我懂你现在的处境——已经能顺利给数据行计算并写入total列，可就是卡在表头那行不知道怎么加total字段对吧？其实用Ruby的CSV库完全可以轻松解决，我给你两种靠谱的实现方式：

方法一：利用CSV的表头识别功能 #

如果你的CSV文件是带表头的格式，Ruby的CSV库可以自动识别表头行，用row.header_row?就能判断当前行是不是表头，直接追加total字段就行：

require 'csv'

# 替换成你的源文件和目标文件路径
source_path = 'input.csv'
output_path = 'output.csv'

CSV.open(output_path, 'w') do |new_csv|
  # 遍历源文件的每一行，开启表头识别
  CSV.foreach(source_path, headers: true) do |row|
    if row.header_row?
      # 给表头追加'total'字段
      new_csv << row.headers + ['total']
    else
      # 计算col_1和col_2的总和（注意转成整数，因为CSV读出来是字符串）
      total = row['col_1'].to_i + row['col_2'].to_i
      # 把当前行数据加上total后写入新文件
      new_csv << row.fields + [total]
    end
  end
end

这里开启了headers: true参数，让CSV库识别表头，这样你还能直接通过列名（比如row['col_1']）获取数据，比用索引更直观。

方法二：手动标记表头行 #

如果你不想依赖表头识别功能，也可以手动判断第一行是表头，处理完后切换标记即可：

require 'csv'

source_path = 'input.csv'
output_path = 'output.csv'
is_first_row = true

CSV.open(output_path, 'w') do |new_csv|
  CSV.foreach(source_path) do |row|
    if is_first_row
      # 处理表头行，追加'total'
      new_csv << row + ['total']
      is_first_row = false
    else
      # 计算总和并写入
      total = row[1].to_i + row[2].to_i
      new_csv << row + [total]
    end
  end
end

这个方法更通用，不管你的CSV有没有明确的表头格式，都能正常工作。

因为你提到源文件没有引号作为分隔符，Ruby的CSV默认就支持这种无引号的简洁格式，所以不用额外设置任何分隔符参数，直接跑代码就行~

内容的提问来源于stack exchange，提问作者Tom Hughes