Разделение маски на другие маски по ее элементам

У меня есть маска массива python произвольного размера, которая включает (0,1,2,3) числа. Я хочу разделить его на 4 подмаски, каждая из которых включает логические значения, первая из которых имеет True в позициях с 0 элементами, вторая имеет True в позиции с одним элементом и так далее. Мой входной массив слишком велик, и мне нужен быстрый способ сделать это на Python. Вот пример:

mask = [[0 1 1]
        [2 0 3]
        [3,3,2]
        [1,1,1]]

Я хочу иметь:

sub_mask[0] = [[1 0 0]
               [0 1 0]
               [0 0 0]
               [0 0 0]]
sub_mask[1] = [[0 1 1]
               [0 0 0]
               [0 0 0]
               [1 1 1]]
sub_mask[2] = [[0 0 0]
               [1 0 0]
               [0 0 1]
               [0 0 0]]
sub_mask[3] = [[0 0 0]
               [0 0 1]
               [1 1 0]
               [0 0 0]]

python arrays python-3.x mask

20.09.2018 16:50

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

Ответы 2

Вы можете использовать numpy:

import numpy as np

mask = [[0, 1, 1],
        [2, 0, 3],
        [3, 3, 2],
        [1, 1, 1]]
mask = np.array(mask)

def get_sub_mask(n):
    # Create an empty sub-mask with all 0's
    sub_mask = np.zeros(mask.shape)
    # Set elements to 1 based on their position in the original mask
    sub_mask[np.where(mask == n)] = 1
    return sub_mask

21.09.2018 00:30

Вы можете сделать это с помощью простого понимания списка:

sub_mask = [
    mask == val
    for val in np.unique(mask)
]

21.09.2018 00:41

Другие вопросы по теме

Выделение массива размером более 2 ГБ в C (или C++)

Сортировка массива строк в соответствии с порядком другого массива с использованием первого столбца

Как массив типов значений размещается в куче?

Каждый n-й элемент из массива, начиная откуда-то в массиве C#

Как сгенерировать массив массивов с динамическим размером с помощью numpy?

Как цикл for выполняет итерацию по массиву?

Как пройти через aara и добавить каждое значение в div с определенным классом

Сохранить результат из SQL в массиве для доставки JSON в Php

Array_filter ключ возврата php, а не последовательность

Выберите столбец mysql (который является массивом), в котором определен другой столбец: - и вставьте его в переменную php

Разделение маски на другие маски по ее элементам

Ответы 2

Другие вопросы по теме

Похожие вопросы