python - 在 Python 中将矩阵元素从 str 转换为 int

Question

我需要将矩阵的元素从 str 转换为 int （在我从文件中读取它们之后）。我不想使用任何外部库。

这是我到目前为止制作的并且正在工作，但我对它不满意（不是很pythonic）：

def str2int(matrix):
    n = 1 
    m = 1 
    if type(matrix) is ListType:
        n = len(matrix)
        if type(matrix[0]) is ListType:
            m = len(matrix[0])
    new_matrix = []
    i = 0 
    while i < n:
        new_matrix.append([])
        j = 0 
        while j < m:
            new_matrix[i].append(int(matrix[i][j]))
            j += 1
        i += 1
    return new_matrix

有更好的想法吗？

谢谢。

Answer 1

使用列表理解：

return [[int(i) for i in row] for row in matrix]