Ruby: Есть что-то вроде Enumerable # drop, который возвращает перечислитель вместо массива? - PullRequest
Новая
       36

Ruby: Есть что-то вроде Enumerable # drop, который возвращает перечислитель вместо массива?

8 голосов
/ 04 февраля 2010

У меня есть несколько больших файлов фиксированной ширины, и мне нужно удалить строку заголовка.

Отслеживание итератора не кажется идиоматичным. 

# This is what I do now.
File.open(filename).each_line.with_index do |line, idx|
  if idx > 0
     ...
  end
end

# This is what I want to do but I don't need drop(1) to slurp
# the file into an array.
File.open(filename).drop(1).each_line do { |line| ... }

Что такое идиома для Ruby?

Ответы [ 6 ]

7 голосов
/ 04 февраля 2010

Это слегка аккуратнее: 

File.open(fname).each_line.with_index do |line, lineno|
  next if lineno == 0
  # ...
end

или 

io = File.open(fname)
# discard the first line
io.gets
# process the rest of the file
io.each_line {|line| ...}
io.close
5 голосов
/ 04 февраля 2010

Если вам это нужно более одного раза, вы можете написать расширение Enumerator. 

class Enumerator
  def enum_drop(n)
    with_index do |val, idx|
      next if n == idx
      yield val
    end
  end
end

File.open(testfile).each_line.enum_drop(1) do |line|
  print line
end

# prints lines #1, #3, #4, …
2 голосов
/ 04 февраля 2010

Теперь, когда вы получили разумные ответы, вот совершенно другой способ справиться с этим. 

class ProcStack
  def initialize(&default)
    @block = default
  end
  def push(&action)
    prev = @block
    @block = lambda do |*args|
      @block = prev
      action[*args]
    end
    self
  end
  def to_proc
    lambda { |*args| @block[*args] }
  end
end
#...
process_lines = ProcStack.new do |line, index|
  puts "processing line #{index} => #{line}"
end.push do |line, index|
  puts "skipping line #{index} => #{line}"
end
File.foreach(filename).each_with_index(&process_lines)

С первого раза это не идиоматично, не ужасно интуитивно, но весело!

1 голос
/ 09 апреля 2013

Я думаю, что вы находитесь на правильном пути с помощью Enumerator и пропустите (1). По какой-то странной причине, хотя Enumerable определяет #drop, Enumerator - нет. Вот рабочий перечислитель # drop: 

  class Enumerator
    def drop(n_arg)
      n = n_arg.to_i # nil becomes zero
      raise ArgumentError, "n must be positive" unless n > 0
      Enumerator.new do |yielder|
        self.each do |val|
          if n > 0
            n -= 1
          else
            yielder << val
          end
        end
      end
    end
  end
1 голос
/ 04 февраля 2010

Я сомневаюсь, что это идиоматично, но это просто. 

f = File.open(filename)
f.readline
f.each_line do |x|
   #...
end
1 голос
/ 04 февраля 2010

С макушки головы, но я уверен, что после дальнейших исследований есть более элегантный способ 

File.open( filename ).each_line.to_a[1..-1].each{ |line|... }

Хорошо, поцарапайте, что ... провели небольшое исследование, и это могло бы быть лучше 

File.open( filename ).each_line.with_index.drop_while{ |line,index|  index == 0 }.each{ |line, index| ... }
...