Ruby pobierające elementy ukośne w tablicy 2d

Question

Próbowałem pewnych problemów z moją tablicą 2D ruby, a mój LOC znacznie się zmniejsza, gdy robię cięcie w tablicy. Na przykład:

require "test/unit" 

class LibraryTest < Test::Unit::TestCase 

    def test_box 
    array = [[1,2,3,4],[3,4,5,6], [5,6,7,8], [2,3,4,5]] 
    puts array[1][2..3] # 5, 6 
    puts array[1..2][1] # 5, 6, 7, 8 
    end 
end 

Chcę wiedzieć, czy istnieje sposób na uzyskanie przekątnej plasterka? Powiedzmy, że chcę zacząć od [0,0] i chcę przekątnej plastra 3. Wtedy dostanę elementy od [0,0], [1,1], [2,2] i otrzymam tablicę [1,4,7] na przykład powyżej. Czy istnieje jakiś magiczny, jednolinijkowy kod rubowy, który może to osiągnąć? 3x zrobić {jakieś magiczne rzeczy?}

Answer 1

puts (0..2).collect { |i| array[i][i] } 

Answer 2

Lepiej może być jedno-liner, który wykorzystuje bibliotekę matrix:

require 'matrix' 
Matrix.rows(array).each(:diagonal).to_a 

Answer 3

Ruby fragment oparty off Get all the diagonals in a matrix/list of lists in Python

ten ma weź wszystkie przekątne. Tak czy inaczej, chodzi o to, aby umieścić tablicę z różnych stron, tak aby przekątne układały się w rzędach i kolumnach:

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

# pad every row from down all the way up, incrementing the padding. 
# so: go through every row, add the corresponding padding it should have. 
# then, grab every column, that’s the end result. 

padding = arr.size - 1 
padded_matrix = [] 

arr.each do |row| 
    inverse_padding = arr.size - padding 
    padded_matrix << ([nil] * inverse_padding) + row + ([nil] * padding) 
    padding -= 1  
end 

padded_matrix.transpose.map(&:compact)