在ruby中选择二维数组的列方式

时间:2014-05-27 21:15:08

标签: ruby multidimensional-array

我有一个二维数组A = [[a1,a2,a3],[b1,b2,b3],[c1,c2,c3]]. 我想以列方式访问此数组。 类似的东西 -

A[all][0]
-> [a1,b1,c1]

我该怎么做?

4 个答案:

答案 0 :(得分:9)

使用#transpose方法执行以下操作:

A.transpose.each do |ary|
   # your code
end

根据您的评论,我建议使用Matrix课程。创建Matrix对象后,您可以访问它的元素,行式列式

require 'matrix'

A = [['a1','a2','a3'],['b1','b2','b3'],['c1','c2','c3']]

mat = Matrix[ *A ]
mat.column(1).to_a # => ["a2", "b2", "c2"]

答案 1 :(得分:6)

另一种选择是使用Array#map

A = [["a1","a2","a3"],["b1","b2","b3"],["c1","c2","c3"]]
=> [["a1", "a2", "a3"], ["b1", "b2", "b3"], ["c1", "c2", "c3"]]
>> col = 0
=> 0
>> A.map{|a| a[col]}
=> ["a1", "b1", "c1"]

可根据需要转换为方法。

答案 2 :(得分:0)

我会使用Array#transpose,但这是使用Array#zip的替代方法:

A = [[1,2,3],[4,5,6],[7,8,9]]

A.first.zip(*A[1..-1]).first #=> [1, 4, 7]

相反,如果

a = [[1,2,3],[4,5,6],[7,8,9]]

你并不介意改变a,你可以这样做:

a.shift.zip(*a).first #=> [1, 4, 7]

答案 3 :(得分:0)

简短答案:

A_t = A.tranpose
A_t[0]