Руководство по библиотеке Ruby CSV, часть II

 guests = CSV.read('guests.csv', headers:true) guests['Times arrived'] #=> ["34", "55", "22", "34"] guests[0] #=> #<CSV::Row "Name":"Carlos" "Times arrived":"22" "Total $ spent":"4352" "Food feedback":"I am \"pleased\", but could be better"> 

Мы даже можем объединить эти 2 обозначения для доступа только к определенной ячейке:

 guests['Times arrived'][1] #=> Returns '55' guests[1]['Times arrived'] #=> Also returns 55 

Ruby с этим кодом говорит следующее: «Дайте мне значение ячейки, которая находится под столбцом« Время пришло »во второй строке». Ухоженная!

Практическое использование заголовков CSV

Давайте попробуем взять наш файл guest.csv и изменить столбец «Всего $ потрачено», чтобы он содержал десятичные числа вместо целых чисел:

 new_guests_csv = [] # We create an array to gold the new CSV data CSV.foreach('guests.csv',headers:true) do |guest| # Iterate over each row of our CSV file guest['Total $ spent'] = guest['Total $ spent'].to_f # new_guests_csv << guest # Add the new row into new_guests_csv end 

Я решил использовать метод .foreach для чтения из файла здесь. Я мог бы также использовать CSV#read :

 new_guests_csv = [] # We create an array to gold the new CSV data old_guests_csv = CSV.read('guests.csv', headers:true) # Reads the entire content of the CSV into the variable old_guests_csv.each do |guest| # old_guests_csv is CSV::Table object which has #each method to iterate over its rows guest['Total $ spent'] = guest['Total $ spent'].to_f # Same thing as with our previous code new_guests_csv << guest # Add the new row into new_guests_csv end 

Мы могли бы сделать это, если мы хотим сохранить наши данные в новом файле:

 CSV.open('updated_guests.csv', 'w') do |csv| # Create a new file updated_guests.csv csv << ['Name', 'Times arrived', 'Total $ spent (decimals)', 'Food feedback'] # Add new headers new_guests_csv.each do |row| # Since we now have the entire updated CSV file in this variable as a double array, # we iterate over each (array) element csv.puts row end end 

Часто при записи данных в новый CSV-файл вам может понадобиться изменить имена заголовков. Большинство людей сначала добавляют новые заголовки (как мы это сделали во второй строке выше), а затем перебираем измененную переменную (в нашем случае new_guests_csv ) и добавляем каждую строку отдельно.

Подсказка: что если у вас есть файл с двумя одинаковыми именами столбцов? Предположим, что наш файл guest.csv содержит еще один столбец «Имя», как мы будем к нему обращаться? Как вы уже знаете, доступ к первому столбцу можно получить с помощью row_object['Name'] . Чтобы получить доступ ко второму, напишите row_object['Name', 2]

Как читать БОЛЬШОЙ файл CSV, не теряя здравомыслие

Язык Ruby в первую очередь оптимизирован для людей, а не для компьютеров. Таким образом, производительность является вторичной, а удобство — первым в списке приоритетов. К сожалению, это может быть проблемой, когда вы пытаетесь что-то вроде чтения большого файла в память. Если размер вашего файла составляет 300 мегабайт, а на компьютере, на котором выполняется ваш код, скажем, 512 МБ ОЗУ, возникнет проблема.

Вернемся к дяде Бобу. Перенесемся в будущее, у дяди Боба есть франшиза и многомиллионный бизнес. Он все еще хранит свой CSV-файл, хотя теперь в нем более 70 000 строк! Если мы попытаемся прочитать и манипулировать этим файлом с помощью CSV.read (который будет считывать ВЕСЬ файл в память), размер нашей программы в ОЗУ увеличится в несколько раз. Это происходит потому, что в памяти есть объект (массив CSV::Row если у нас есть заголовки) для каждой строки нашего CSV-файла. Это 70 000 объектов! Допустим, мы хотим вернуть только те строки с людьми, которые приходили в ресторан более 10 раз:

 #...doing some regular stuff, the program memory size is 12MB guests = CSV.read('guests.csv',headers:true) # a big CSV file is read into memory, the size is now over 100MB! guests.select do |row| row['Times arrived'].to_i > 10 # if we put the return value of the select block in another variable, we'll have even bigger size end 

Конечно, мы можем избежать этого с помощью CSV.foreach, который вместо чтения всего файла в память будет повторять строку за строкой и не будет влиять на общий размер памяти программы после завершения итерации. Но здесь есть небольшая проблема, нам придется реструктурировать нашу программу:

 guests_who_visited_more_than_ten_times = Array.new CSV.foreach('guests.csv', headers:true) do |guest| guests_who_visited_more_than_ten_times << guest if guest['Times arrived'] > 10 end 

Здесь мы обмениваем выразительность на эффективность, первый пример, когда мы читали из памяти, был лучше, потому что наши данные были реальным объектом, на котором мы могли использовать Array#select . К счастью, вы можете получить лучшее из обоих миров, сохраняя память и выразительность, используя перечислитель ( опора для Avdi из RubyTapas, где я выучил этот удивительный трюк):

 CSV.open('guests.csv', headers:true) do |guest| guests = guest.each guests.select do |row| row['Times arrived'].to_i > 10 end end 

Если вы не знакомы со второй строкой, мы в основном создаем объект перечислителя и сохраняем его в переменной guest. Перечислители позволяют нам выполнять итерацию «по требованию», скажем, если бы у меня был код, который насчитал от 1 до 100:

 1.upto(10) { |x| px } 

Превращение этого объекта в объект перечислителя позволит мне, например, считать от 1 до 5 в одной части моей программы, а затем закончить отсчет до 10:

 enum = 1.upto(10) p enum.next #=> 1 p 'bunch of other code here' p enum.next #=> 2 p 'and here' p enum.next #=> 3 

Характер «по требованию» счетчиков делает их дружественными к памяти. По сравнению с первым примером с использованием CSV.read у нас не будет тысяч объектов в оперативной памяти, и в отличие от CSV.foreach , нам не нужно повторно CSV.foreach код путем изменения логики.

Ну, это конец серии Ruby CSV. Я надеюсь, что вы многому научились!