Учитывая целочисленную матрицу aи неотрицательное целое число i, выведите отображение, bкоторое отображает различные значения в iстолбце th aна строки a, имеющие это значение в iстолбце th.

Вы можете предположить, что он iнаходится в полуоткрытом диапазоне [0, num_cols(a))(или [1, num_cols(a)]если вы решите использовать индексы на основе 1), и что все целые числа находятся в пределах представимого диапазона для вашего языка. Ввод и вывод могут быть выполнены любым разумным способом, при условии, что он удовлетворяет основным требованиям задачи (2D-массив -> отображение из целых чисел в двумерные массивы целых). До тех пор, пока отображение ясное и последовательное, ключи не нужно включать в вывод.

Примеры

[[1]], 0 -> {1: [[1]]}
[[3, 4, 5], [1, 4, 2], [5, 5, 5], [7, 7, 7], [1, 5, 9]], 1 -> {4: [[3, 4, 5], [1, 4, 2]], 5: [[5, 5, 5], [1, 5, 9]], 7: [[7, 7, 7]]}
[[1, 2, 3, 4, 5], [5, 4, 3, 2, 1], [2, 3, 4, 5, 6], [8, 9, 100, 0, 2]], 4 -> {5: [[1, 2, 3, 4, 5]], 1: [[5, 4, 3, 2, 1]], 6: [[2, 3, 4, 5, 6]], 2: [[8, 9, 100, 0, 2]]}

Это код-гольф , поэтому выигрывает самый короткий ответ в байтах.

Октава , 24 байта

@(a,i)@(n)a(a(:,i)==n,:)

Попробуйте онлайн!

Это создает анонимную функцию, которая возвращает матрицу, строки которой соответствуют критериям. Массивы октавных индексов равны 1, а не нулю, а строки матрицы разделены знаком a ;.

Матрицы - это то, что Octave делает лучше всего - настолько хорошо, что фактически эта задача может быть решена с использованием чистого синтаксиса, без встроенных функций.

объяснение

@(a,i)                   % creates an anonymous function that...
      @(n)               % returns another function that takes input n and
                         % maps it to the rows of a.
          a(         ,:) % Return all the columns of a, with the rows filtered by...
            a(:,i)       % whether the ith column of each row of a...
                  ==n    % equals n

Рубин , 26 байт

->a,i{a.group_by{|x|x[i]}}

Попробуйте онлайн!

Wolfram Language (Mathematica) , 21 байт

#~GroupBy~Extract@#2&

1-индексироваться. ВозвращаетAssociation сопоставление.

Попробуйте онлайн!

Это редкий случай, когда более длинная функция ( Extract) уменьшает количество байтов (более короткое существо Partили [[ ... ]]), потому что Extractможет карри. Результат - это чрезвычайно краткое решение с двумя функциями.

объяснение

Extract@#2

Функция, извлекающая <second input>элемент th.

#~GroupBy~ ...

Сгруппируйте <first input>в списки, связанные с различными ключами <above function>[element].

Haskell , 64 60 байт

import Data.List
i!l=[(k,[a|a<-l,a!!i==k])|k<-nub$map(!!i)l]

Попробуйте онлайн!

Чисто , 40 байт

import StdEnv

\n l i=filter(\a=a!!n==i)l

Попробуйте онлайн!

Лямбда ( :: Int [[Int]] Int -> [[Int]]), где частичное применение только первых двух аргументов дает отображение на третий аргумент.

J , 16 байт

-3 байта благодаря FrownyFrog!

{"1(~.@[;"0</.)]

Попробуйте онлайн!

Объяснение:

Диадический глагол, принимающий iза левый аргумент и aза правый.

] это правильный аргумент, a

{"1находит числа в iстолбце th в каждой строке

</. блокирует группы из правого аргумента, выбираемого клавишами, предоставленными левым

~.@[ находит уникальные ключи

;"0 связывает ключи с выбранными группами

JQ, 100 байт

использует объект для вывода, принимает аргумент командной строки $fплюс массив на стандартном вводе

([.[]|.[$f]]|unique) as $c|[$c[] as $d|{($d|tostring):([.[]|[select(.[$f]==$d)]]|add)}]|add

deobfuscated:

.fieldnum as $field |
.input as $input |
([$input[] | .[$field]] | unique) as $categories |
[
    $categories[] as $category |
    {
        ($category | tostring) :
            ([$input[] | [select(.[$field]==$category)]] | add)
    }
] | add

R , 79 55 байтов

function(a,i)for(z in unique(a[,i]))print(a[a[,i]==z,])

Попробуйте онлайн!

@JayCe сбрил 24 байта

Python 3 , 45 байт

lambda a,i:lambda n:[r for r in a if r[i]==n]

Попробуйте онлайн!

Возвращает отображение, представленное как анонимная лямбда.

Протон , 29 байт

a=>i=>n=>filter(r=>r[i]==n,a)

Попробуйте онлайн!

-3 байта благодаря г-ну Xcoder, использующему карри и filter(TBH я немного удивлен, что filterфактически работал)

JavaScript (Node.js) , 29 байт

a=>i=>n=>a.filter(e=>e[i]==n)

Попробуйте онлайн!

Обновленный теперь, когда я понимаю свободные требования к продукции. Это использует карри в качестве техники игры в гольф, а также возвращает функцию, которая принимает входные данные nи отображает их в соответствующие массивы.

Желе , 5 байт

ịⱮ⁹¹ƙ

Попробуйте онлайн!

Опускает ключи, но должно быть ясно.

Аргумент 1: я + 1
Аргумент 2: а

Java 10, 135 64 байта

m->i->n->new java.util.Stack(){{for(var a:m)if(a[i]==n)add(a);}}

Возвращает Function<Integer, List<int[]>>принимающий целочисленный ввод n, который возвращает список массивов (строк матрицы), где i'-ые значения равны заданному n.

Попробуйте онлайн.

Объяснение:

m->i->               // Method with int-matrix and int parameters and Function return-type
  n->                //  Return a Function with integer as parameter
    new java.util.Stack(){{
                     //  and List of integer-arrays as return-type
      for(var a:m)   //   Loop over the arrays of the input-matrix
        if(a[i]==n)  //    If the `i`'the value of the current array equals `n`:
          add(a);}}  //     Add it to the return-List