我将如何迭代地将我的数组输入转换为整数

Posted 2021-05-06

假设我有以下输入作为多维列表

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

我怎样才能遍历循环并计算每个项目。

答案

# your mapping
m = '.C-|'

# iterate rows then inside iterate columns
out = [[m.index(c) for c in r] for r in grid]

另一答案

您可以按如下方式使用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)