Ruby:正则表达式的过滤器数组？

Question

对我来说,这是一个常见的,重复的习惯用法:使用正则表达式过滤数组,然后返回子数组.我的方法似乎不像Ruby(我来自Java).我最终有很多方法看起来很像这样.

什么是惯用的Ruby方法来改进这段代码？

def get_all_gifs(items_)
  output = Array.new
  filter = /\.jpg$/
  items_.each do |item|
    next if item =~ filter
    output << item
  end
  output
end

Answer 1

如果你想找到所有的GIF:

def get_all_gifs(files)
  files.select{ |i| i[/\.gif$/] }
end

如果你想找到所有的jpegs:

def get_all_jpgs(files)
  files.select{ |i| i[/\.jpe?g$/] }
end

运行它们:

files = %w[foo.gif bar.jpg foo.jpeg bar.gif]
get_all_gifs(files) # => ["foo.gif", "bar.gif"]
get_all_jpgs(files) # => ["bar.jpg", "foo.jpeg"]

可是等等!还有更多!

如果您想按类型对它们进行分组,然后根据扩展名进行提取,该怎么办？:

def get_all_images_by_type(files)
  files.group_by{ |f| File.extname(f) }
end

这是文件的类型:

get_all_images_by_type(files).keys # => [".gif", ".jpg", ".jpeg"]

以下是获取特定类型的方法:

get_all_images_by_type(files) # => {".gif"=>["foo.gif", "bar.gif"], ".jpg"=>["bar.jpg"], ".jpeg"=>["foo.jpeg"]}
get_all_images_by_type(files)['.gif'] # => ["foo.gif", "bar.gif"]
get_all_images_by_type(files).values_at('.jpg', '.jpeg') # => [["bar.jpg"], ["foo.jpeg"]]

Answer 2

看看Enumerable.grep,它是一种非常强大的查找/过滤任何可枚举事物的方法.

Answer 3

$ cat foo.rb
images = %w[foo.gif foo.png foo.jpg bar.jpeg moo.JPG]
jpgs = images.select{|e| e =~ /\.jpe?g$/i}
puts jpgs.inspect

$ ruby foo.rb
["foo.jpg", "bar.jpeg", "moo.JPG"]

对正则表达式的更改是为了除了"jpg"之外你还可以匹配"jpeg",无论大小写如何.

Answer 4

images = %w[foo.gif foo.png foo.jpg bar.jpeg moo.JPG]
images.grep(/.jpg/i) # => ["foo.jpg", "moo.JPG"]

Answer 5

对于这个特定问题，我们可以不使用正则表达式。用String#end_with?

images = %w[foo.gif foo.png foo.jpg bar.jpeg moo.JPG]
images.find_all{|e| e.end_with?(".jpg",".jpeg")}
# => ["foo.jpg", "bar.jpeg"]