在Python中將矩陣元素從str轉換爲int

Question

我需要將矩陣的元素從str轉換爲int（在我從文件中讀取它們之後）。 我不想使用任何外部庫。

這是我製作的，到目前爲止，它的工作，但我不喜歡它（不是很符合Python）：

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

使用list comprehension：

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