Ruby：通过正则表达式过滤数组

发布于06月28日

对我来说，这是一个常见且重复的习惯用法:使用正则表达式过滤数组，然后返回子array.我的方法看起来不太像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

推荐答案

如果你想找到所有的GIF:

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

如果要查找所有JPEG:

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"]]