Question

我有一个数组：

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

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

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 `<main>'

应该是：

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

如何转置不同大小的红宝石阵列？

3 个答案: