如何在Ruby中检查CSV列数据并按条件提取同行对应数据？

嘿，我来帮你搞定这两个CSV处理需求！Ruby自带的CSV标准库就足够应付这些场景，不用额外装依赖，下面直接上代码和说明：

1. 检查指定列并提取对应行数据 #

假设你的CSV文件结构类似这样（以data.csv为例）：

Name,Age,Level,Department
Alice,28,2,Engineering
Bob,35,3,Marketing
Charlie,22,1,Design

如果要提取Age列大于30的所有行，或者Department等于"Engineering"的行，可以这么写：

require 'csv'

# 定义要检查的列和判断条件
target_column = "Age"
condition = ->(value) { value.to_i > 30 }

matching_rows = []

CSV.foreach('data.csv', headers: true) do |row|
  # 获取指定列的值，注意列名要和CSV里的完全一致（大小写敏感）
  column_value = row[target_column]
  # 满足条件就把整行数据转成哈希加入结果数组
  matching_rows << row.to_h if condition.call(column_value)
end

# 输出提取到的行
puts "符合条件的行："
matching_rows.each { |row| puts row.inspect }

如果需要多列联合判断，直接修改condition里的逻辑就行，比如value.to_i > 25 && row["Level"].to_i >=2。

2. 按条件筛选Level列并输出对应的severity值 #

首先我们先定义Level与severity的对应关系（你可以根据实际需求调整）：

• Level 1 → "low"
• Level 2 → "medium"
• Level 3 → "high"

接下来筛选符合条件的Level（比如Level >=2），并输出对应的severity：

require 'csv'

# 定义Level与severity的映射哈希
level_severity = {
  "1" => "low",
  "2" => "medium",
  "3" => "high"
}

# 遍历CSV，筛选Level >=2的行并输出对应severity
CSV.foreach('data.csv', headers: true) do |row|
  level = row["Level"]
  # 跳过Level为空或者不符合条件的行
  next unless level && level.to_i >= 2

  # 处理未定义的Level值，避免报错
  severity = level_severity[level] || "unknown"
  puts "#{row['Name']}对应的severity为：#{severity}"
end

运行这段代码后，会输出所有符合条件的行对应的severity值，要是遇到不在映射里的Level，会显示"unknown"，避免程序崩溃。

如果你的CSV路径、列名或者判断条件有变化，直接替换代码里对应的部分就好，非常灵活～

内容的提问来源于stack exchange，提问作者Joshua Ng