quick script to change a csv no problemo

csv_manip.rb

require 'CSV'
puts "lets go ruby"

def fruit_we_want_to_change? fruit
  fruits_affected = [
    'apple',
    'banana',
  ]

  fruits_affected.include?(fruit)
end

CSV.open("output.csv", "wb") do |csv_out|

  CSV.foreach("sample_data.csv") do |row|
    fruit = row[0]
    location = row[2]

    if fruit_we_want_to_change?(fruit)
      new_location = 'sainsburys'

      puts 'changing ' + fruit + ' location to ' + new_location
      row[2] = new_location
    end

    csv_out << row
  end
end


# SAMPLE CSV:

# fruit,some irrelevant data,location
# apple,blah,england
# pineapple,blah,brazil
# banana,blah,india
# banana,blah,west indies
# banana,blah,antarctic
# mango,blah,pakistan