Ruby - 数组展平

Question

我有一个 n 深度数组，其中 n 是一个大于或等于 2 的变量:

[[1,1],[[1,1],[1,1]]]

我想将这个数组展平为恰好有 2 个深度，如下所示:

[[1,1],[1,1],[1,1]]

谁能想出一个好的方法来实现它？

Answer 1

应该这样做。

def flatten_after_first(arr)
  arr.flat_map { |a| a.first.is_a?(Array) ? a.map(&:flatten) : [a] }
end

flatten_after_first [[1,1],[[1,1],[1,1]]]
  #=> [[1, 1], [1, 1], [1, 1]]

flatten_after_first [[1,1], [[2,2], [2,2]], [[[3,3], [3,3]], [[3,3], [3,3]]]] 
  #=> [[1, 1], [2, 2], [2, 2], [3, 3, 3, 3], [3, 3, 3, 3]]