CheckIO: Pattern Recognition

import itertools as it


def checkio(pattern, image):
    pattern_height, pattern_width = len(pattern), len(list(zip(*pattern)))
    image_height, image_width = len(image), len(list(zip(*image)))
    scan = lambda: it.product(range(pattern_height), range(pattern_width))

    for i, j in it.product(range(image_height), range(image_width)):
        try:
            if all(image[i + x][j + y] == pattern[x][y] for (x, y) in scan()):
                for x, y in scan():
                    image[i + x][j + y] += 2
        except IndexError:
            continue
    return image


#These "asserts" using only for self-checking and not necessary for auto-testing
if __name__ == '__main__':
    assert checkio([[1, 0], [1, 1]],
                   [[0, 1, 0, 1, 0],
                    [0, 1, 1, 0, 0],
                    [1, 0, 1, 1, 0],
                    [1, 1, 0, 1, 1],
                    [0, 1, 1, 0, 0]]) == [[0, 3, 2, 1, 0],
                                          [0, 3, 3, 0, 0],
                                          [3, 2, 1, 3, 2],
                                          [3, 3, 0, 3, 3],
                                          [0, 1, 1, 0, 0]]
    assert checkio([[1, 1], [1, 1]],
                   [[1, 1, 1],
                    [1, 1, 1],
                    [1, 1, 1]]) == [[3, 3, 1],
                                    [3, 3, 1],
                                    [1, 1, 1]]
    assert checkio([[0, 1, 0], [1, 1, 1]],
                   [[0, 0, 1, 0, 0, 0, 0, 0, 1, 0],
                    [0, 1, 1, 1, 0, 0, 0, 1, 1, 1],
                    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                    [0, 0, 0, 0, 0, 1, 0, 0, 0, 0],
                    [0, 1, 0, 0, 1, 1, 1, 0, 1, 0],
                    [1, 1, 1, 0, 0, 0, 0, 0, 1, 1],
                    [0, 0, 0, 1, 1, 1, 0, 0, 0, 0],
                    [0, 0, 1, 0, 0, 0, 0, 1, 0, 0],
                    [0, 1, 1, 0, 0, 0, 1, 1, 1, 0],
                    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]) == [[0, 2, 3, 2, 0, 0, 0, 2, 3, 2],
                                                         [0, 3, 3, 3, 0, 0, 0, 3, 3, 3],
                                                         [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                                                         [0, 0, 0, 0, 2, 3, 2, 0, 0, 0],
                                                         [2, 3, 2, 0, 3, 3, 3, 0, 1, 0],
                                                         [3, 3, 3, 0, 0, 0, 0, 0, 1, 1],
                                                         [0, 0, 0, 1, 1, 1, 0, 0, 0, 0],
                                                         [0, 0, 1, 0, 0, 0, 2, 3, 2, 0],
                                                         [0, 1, 1, 0, 0, 0, 3, 3, 3, 0],
                                                         [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]

import itertools as it

def checkio(pattern, image):

pattern_height, pattern_width = len(pattern), len(list(zip(*pattern)))

image_height, image_width = len(image), len(list(zip(*image)))

scan = lambda: it.product(range(pattern_height), range(pattern_width))

for i, j in it.product(range(image_height), range(image_width)):

try:

if all(image[i + x][j + y] == pattern[x][y] for (x, y) in scan()):

for x, y in scan():

image[i + x][j + y] += 2

except IndexError:

continue

return image

#These "asserts" using only for self-checking and not necessary for auto-testing

if __name__ == '__main__':

assert checkio([[1, 0], [1, 1]],

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

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

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

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

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

[0, 3, 3, 0, 0],

[3, 2, 1, 3, 2],

[3, 3, 0, 3, 3],

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

assert checkio([[1, 1], [1, 1]],

[[1, 1, 1],

[1, 1, 1],

[1, 1, 1]]) == [[3, 3, 1],

[3, 3, 1],

[1, 1, 1]]

assert checkio([[0, 1, 0], [1, 1, 1]],

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

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

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

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

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

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

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

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

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

[0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]) == [[0, 2, 3, 2, 0, 0, 0, 2, 3, 2],

[0, 3, 3, 3, 0, 0, 0, 3, 3, 3],

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

[0, 0, 0, 0, 2, 3, 2, 0, 0, 0],

[2, 3, 2, 0, 3, 3, 3, 0, 1, 0],

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

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

[0, 0, 1, 0, 0, 0, 2, 3, 2, 0],

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

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

https://www.checkio.org/mission/matrix-pattern/

ats-min.com

Just another programmer's site

CheckIO: Pattern Recognition

コメントを残すコメントをキャンセル

ats-min.com

最近の投稿

最近のコメント

アーカイブ

メタ情報

ats-min.com

Just another programmer's site

CheckIO: Pattern Recognition

Previous post

Next post

コメントを残す コメントをキャンセル

ats-min.com

最近の投稿

最近のコメント

アーカイブ

メタ情報

コメントを残すコメントをキャンセル