Question

我创建了一个脚本，允许我将CSV作为多维数组导入，在数组的每一行中搜索，如果搜索字符串与行中的任何元素匹配，则将整行返回到新数组，写作：

require 'csv'
array = CSV.read('CSVlist.csv') #=> [["Name1", "8675309"],["Name2", "5557891"], ["Name3", "5557890"]]
shifted_array = Array.new
results_array = Array.new

while array.empty? == false
    shifted_array = array.shift 
    shifted_array.each do |f|
        if f =~ /8675309/  
           shifted_array = shifted_array.join(",") #=> ["Name1,8675309"]
           results_array.push(shifted_array)
        end
    end
end
puts results_array #=> ["Name1,8675309"]

以上工作正常。但是，我不是只搜索一个字符串（/8675309/以上），而是想搜索数组并返回另一个数组中的任何元素（这是一个导入的文本文件）。我创建了比较数组，但是如何在主数组（array）上搜索另一个数组（numbers_array，而不是字符串）？

require 'csv'
array = CSV.read('CSVlist.csv') #=> [["Name1", "8675309"],["Name2", "5557891"], ["Name3", "5557890"]]
shifted_array = Array.new
results_array = Array.new
numbers_array = File.readlines("list1.txt").map &:split #=> ["5551234", "5557890", "8675309"]

while array.empty? == false
    shifted_array = array.shift
    shifted_array.each do |f|
        if f =~ ???? # want the search to compare f to any element in numbers_array  
            shifted_array = shifted_array.join(",")
           results_array.push(shifted_array)
        end
    end
end
puts results_array  #=> desired output is ["Name1,8675409", "Name3,5557890"]

Answer 1

您可以结合使用.select，.any?，.include?和 .map，它们看起来像是：

results_array = array.select{ |row|
  row.any?{ |e| numbers_array.include? e }
}.map{ |row|
  row.join(',')
}

此代码首先选择仅array的行{em>包含 numbers_array中任何元素的行，然后映射每个join。

使用该语句，您的整个代码块可以缩短为：

require 'csv'

array = CSV.read('CSVlist.csv')
numbers_array = File.readlines('list1.txt').map(&:split)

results_array = array.select{ |row|
  row.any?{ |e| numbers_array.include? e }
}.map{ |row|
  row.join(',')
}

puts results_array

针对另一个数组的所有元素搜索数组元素

1 个答案: