如何转置不同大小的rubyarrays？

我有一个数组：

arr=[[1,2,3],[4,5],[6]], 

我有以下代码：

 arr.transpose 

但它不起作用，如何解决？

我正进入（状态

  [[1,2,3],[4,5],[6]].transpose IndexError: element size differs (2 should be 3) from (irb):13:in `transpose' from (irb):13 from /home/durrant 

我的解决方案

 arr.reduce(&:zip).map(&:flatten) 

输出：

 [[1, 4, 6], [2, 5, nil], [3, nil, nil]] 

Answer 1

一小时前发布了一个类似的答案（但已删除）：

 arr = [[1, 2, 3], [4, 5], [6]] arr[0].zip(*arr[1..-1]) #=> [[1, 4, 6], [2, 5, nil], [3, nil, nil]] 

以上相当于：

 [1, 2, 3].zip([4, 5], [6]) 

这种方法假设您的第一个子arrays总是最长的。 否则结果将被截断：

 arr = [[1, 2], [3, 4, 5], [6]] arr[0].zip(*arr[1..-1]) #=> [[1, 3, 6], [2, 4, nil]] missing: [nil, 5, nil] 

Answer 2

在Stefan的答案中使用zip是最直接的，但如果你坚持使用transpose ，那么：

 l = arr.map(&:length).max arr.map{|e| e.values_at(0...l)}.transpose # => [[1, 4, 6], [2, 5, nil], [3, nil, nil]] 

或者不使用任何一个：

 Array.new(arr.map(&:length).max){|i| arr.map{|e| e[i]}} # => [[1, 4, 6], [2, 5, nil], [3, nil, nil]] 

Answer 3

如果子数组的长度不匹配，则引发IndexError。

 irb(main):002:0> arr=[[1,2,3],[4,5],[6]] => [[1, 2, 3], [4, 5], [6]] irb(main):003:0> arr.transpose IndexError: element size differs (2 should be 3) from (irb):3:in `transpose' from (irb):3 from /Users/liuxingqi/.rvm/rubies/ruby-2.1.2/bin/irb:11:in `
' 

应该：

 irb(main):004:0> arr=[[1,2,3],[4,5,6]] => [[1, 2, 3], [4, 5, 6]] irb(main):005:0> arr.transpose => [[1, 4], [2, 5], [3, 6]] 

要么

 irb(main):006:0> arr=[[1,2],[3,4],[5,6]] => [[1, 2], [3, 4], [5, 6]] irb(main):007:0> arr.transpose => [[1, 3, 5], [2, 4, 6]]