python - 如何在python中通过文本文件获取输入？

Question

在此处输入代码

def prims():

  r = 4;
  c = 4;
  total = 0
  matrix = [ [0 for x in range(r)] for y in range(c)]
  min = 999
  u = 0
  v = 0
  visited = [None]*4
  for i in range(0,4):
    visited[i] = 0
    for j in range(0,4):
        matrix[i][j] = input()
        if matrix[i][j]== 0:
            matrix[i][j] = 999
 visited[0] = 1
 for counter in range(0,3):
    min = 999
    for  i in range(0,4):
        if visited[i] == 1:
            for j in range(0,4):
                if visited[j] != 1:
                    if min > matrix[i][j]:
                        min = matrix[i][j]
                        a = u = i
                        b = v = j
    visited[v] = 1
    total = total + min

    print("edge found :{}->{}:{}".format(u,v,min))
print("The weight of minimum spanning tree is : {}".format(total))
return

prims()

如果我想将输入作为文本文件提供给该程序，我必须做什么。我创建了一个名为“input.txt”的文件。在那里我放置了矩阵的输入。谁能帮我解决一下吗？

输入.txt:

0 28 999 999 999 10 999

28 0 16 999 999 999 14

999 16 0 12 999 999 999

999 999 12 0 22 999 18

999 999 999 22 0 25 24

10 999 999 999 25 0 999

999 14 999 18 24 999 999

Answer 1

这是从文件中读取的方法

file = open('path of input file')
for line in file.readlines():
  print(list(map(int, line.split(' '))))
file.close()

您也可以将其存储到数组中

arr = []
file = open('path of input file')
for line in file.readlines():
  arr.append(list(map(int, line.split(' '))))
print(arr)
file.close()

输出为

[[0, 28, 999, 999, 999, 10, 999], [28, 0, 16, 999, 999, 999, 14], [999, 16, 0, 12, 999, 999, 999], [999, 999, 12, 0, 22, 999, 18], [999, 999, 999, 22, 0, 25, 24], [10, 999, 999, 999, 25, 0, 999], [999, 14, 999, 18, 24, 999, 999]]