как бы я итеративно перевести мой массив ввода в целые числа - PullRequest
0 голосов
/ 14 апреля 2019

Предположим, у меня есть следующий вход в виде многомерного списка

chain = [['.','.','.','.','.','.','.']
        ['.','.','.','.','.','.','.'],
        ['.','.','.','.','.','.','.']]

как я могу пройти по циклу и посчитать каждый элемент.

Ответы [ 2 ]

1 голос
/ 14 апреля 2019
# your mapping
m = '.C-|'

# iterate rows then inside iterate columns
out = [[m.index(c) for c in r] for r in grid]
0 голосов
/ 14 апреля 2019

Вы можете просто использовать библиотеку numpy следующим образом:

import numpy as np
grid = np.array([['.','.','.','.','.','.','.'],
        ['.','C','-','C','-','C','.'],
        ['.','.','.','|','.','.','.'],
        ['.','.','.','C','.','.','.'],
        ['.','.','.','.','.','.','.'],
        ['.','.','.','.','.','.','.'],
        ['.','.','.','.','.','.','.']])

grid = np.where(grid=='.', '0', grid)
grid = np.where(grid=='C', '1', grid)
grid = np.where(grid=='-', '2', grid)
grid = np.where(grid=='|', '3', grid)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...