Numpy расширяет растровое изображение в растровое изображение - PullRequest
0 голосов
/ 05 июля 2019

У меня есть массив numpy, представляющий растровое изображение, и я хотел бы расширить его до вектора векторов, где каждая запись имеет значение исходного растрового изображения в этой позиции, повторенное n раз.

Позвольте мне привести пример, чтобы объяснить, что я имею в виду:

n = 2
original = [0, 0, 1, 1, 0]

wanted = [
    [0, 0],
    [0, 0],
    [1, 1],
    [1, 1],
    [0, 0]
]

Очевидно, я мог бы сделать это с помощью цикла над массивом, но мне бы очень (очень) хотелось, чтобы он был быстрым, поэтому я надеюсь использовать пустые методы.

1 Ответ

2 голосов
/ 05 июля 2019

Просто используйте функцию zip

wanted = list(zip(*[original] * n))

или используйте numpy.dstack

wanted = np.dstack([original] * n)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...