ruby - 获取二维数组中值的坐标

Question

我想获取存储在数组数组中的对象每次出现的坐标。如果我有一个数组:

array = [["foo", "bar", "lobster"], ["camel", "trombone", "foo"]]

和一个对象"foo"，我想得到:

[[0,0], [1,2]]

以下将执行此操作，但它复杂且丑陋:

array.map
.with_index{
  |row,row_index| row.map.with_index {
    |v,col_index| v=="foo" ? [row_index,col_index] : v
  }
}
.flatten(1).find_all {|x| x.class==Array}

有没有更直接的方法来做到这一点？这是 asked之前，并产生了一个类似的不优雅的解决方案。

Answer 1

这里有一个更优雅的解决方案。我有:

• 在最后使用 flat_map 而不是 flatten
• 使用 .each_index.select 而不是 .map.with_index 然后必须在最后去除非数组，这真的很难看
• 添加缩进

array.flat_map.with_index {|row, row_idx|
  row.each_index.select{|i| row[i] == 'foo' }.map{|col_idx| [row_idx, col_idx] }
}