时间戳

首页 > 解决方案 > 存储在 CSV 文件中 - 红宝石分隔符

ruby - 存储在 CSV 文件中 - 红宝石分隔符

问题描述

我正在尝试将报废练习的结果存储到 CSV 文件中。

当前的 CSV 文件给了我以下输出：

Name of Movie 1

Rating 1

Name of Movie 2 

Rating 2

我想得到以下输出：

Name of Movie 1 Rating 1 

Name of Movie 2 Rating 2

这是我的代码，我想它必须处理行/列分隔符：

require 'open-uri'
require 'nokogiri'
require 'csv'

array = []


for i in 1..10
  url = "http://www.allocine.fr/film/meilleurs//?page=#{i}"
  html_file = open(url).read
  html_doc = Nokogiri::HTML(html_file)


  html_doc.search('.img_side_content').each do |element|
    array << element.search('.no_underline').inner_text
    element.search('.note').each do |data|
      array << data.inner_text
    end
  end
end

puts array


csv_options = { row_sep: ',', force_quotes: true, quote_char: '"' }
filepath    = 'allocine.csv'

CSV.open(filepath, 'wb', csv_options) do |csv|
  array.each { |item| csv << [item] }
end

标签： rubycsvexport-to-excelexport-to-csv

解决方案

我认为这里的问题是您没有将元素正确地推入array变量中。基本上，您的数组最终看起来像这样：

['Movie 1 Title', 'Movie 1 rating', 'Movie 2 Title', 'Movie 2 rating', ...]

您真正想要的是一个数组数组，如下所示：

[
  ['Movie 1 Title', 'Movie 1 rating'],
  ['Movie 2 Title', 'Movie 2 rating'],
  ...
]

一旦正确设置了数组，您甚至不需要在 CSV 选项中指定行分隔符。

以下应该可以解决问题：

require 'open-uri'
require 'nokogiri'
require 'csv'

array = []


10.times do |i|
  url = "http://www.allocine.fr/film/meilleurs//?page=#{i}"
  html_file = open(url).read
  html_doc = Nokogiri::HTML(html_file)


  html_doc.search('.img_side_content').each do |element|
    title = element.search('.no_underline').inner_text.strip
    notes = element.search('.note').map { |note| note.inner_text }
    array << [title, notes].flatten
  end
end

puts array

filepath    = 'allocine.csv'
csv_options = { force_quotes: true, quote_char: '"' }

CSV.open(filepath, 'w', csv_options) do |csv|
  array.each do |item|
    csv << item
  end
end

（我还冒昧地将您的for循环更改为 a times，它更像红宝石；））

推荐阅读