在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]