Нахождение индекса элементов в массиве массивов

Question

У меня есть массив массивов:

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

Я хочу найти индекс (строки и столбца) из элементов со значением 1. Как я могу это сделать? Мне нужны эти значения, чтобы я мог манипулировать ячейками по обе стороны от 1. Я посмотрел на метод find.index, но я не уверен.

Answer 1

a = [ 
    [0,0,0,0], 
    [0,1,0,0], 
    [0,0,0,1], 
    [0,0,0,0] 
].flatten 

a.each.with_index.select{|e, _| e == 1}.map{|_, i| i.divmod(4)} 
# => [[1, 1], [2, 3]] 

Следует понимать как (строка 1, столбец 1) и (строка 2, столбец 3).

Answer 2

Ruby имеет очень гибкий функциональный модуль Enumerable, который входит во многие стандартные коллекции (массивы, хэши, наборы). Вы можете найти документацию at ruby-doc.org.

Хороший один вкладыш решить эту проблему в максимально #rows * 2 #columns время будет следующее:

matrix 
    .each.with_index 
    .inject([]) { |acc, (el,idx)| el.include?(1) ? acc.push([idx, el.index(1)]) : acc } 
# => [[1,1],[2,3]] 

Answer 3

z = [ 
    [0,0,0,0], 
    [0,1,0,0], 
    [0,0,0,1], 
    [0,0,0,0] 
] 

Во-первых, давайте найти соответствующие строки:

rows = z.each_with_index.select { |row, index| row.include? 1}.map(&:last) 
# => [1, 2] 

И затем для каждой строки найдем индекс соответствия 1:

cols = rows.map {|row| z[row].each_with_index.select { |item, index| item == 1}.map(&:last) } 
# => [[1], [3]] 

Если мы хотим, мы можем затем объединить их с помощью zip:

rows.zip(cols) 
# => [[1, [1]], [2, [3]]] 

выше работает, даже если ряд z содержит несколько вхождений 1

Например, если у нас было:

z = [ 
    [1,0,0,1], 
    [0,1,0,0], 
    [0,0,0,1], 
    [1,1,1,0] 
] 

затем

rows = z.each_with_index.select { |row, index| row.include? 1}.map(&:last) 
# => [0, 1, 2, 3] 
cols = rows.map {|row| z[row].each_with_index.select { |item, index| item == 1}.map(&:last) } 
# => [[0, 3], [1], [3], [0, 1, 2]] 
rows.zip(cols) 
# => [[0, [0, 3]], [1, [1]], [2, [3]], [3, [0, 1, 2]]] 

Answer 4

Не то фантазии, но вы могли бы использовать две петли:

ary = [ 
    [0, 0, 0, 0], 
    [0, 1, 0, 0], 
    [0, 0, 0, 1], 
    [0, 0, 0, 0] 
] 

result = [] 
ary.each_with_index do |row, i| 
    row.each_with_index do |value, j| 
    result << [i, j] if value == 1 
    end 
end 
result 
#=> [[1, 1], [2, 3]] 

Answer 5

Рассмотрите возможность использования Matrix класса.

require 'matrix' 

arr = [ 
    [0, 0, 0, 0], 
    [0, 1, 0, 0], 
    [0, 0, 0, 1], 
    [0, 0, 0, 0] 
]  

target = 1 

Matrix[*arr].each_with_index.with_object([]) { |(e,row,col),a| 
    a << [row,col] if e==target } 
    #=> [[1, 1], [2, 3]] 

Мне нравится, как это читается.