如何在python中保存文件内容？

Question

我刚刚开始研究python,这是一个非常基本的问题:

我有一个input.txt包含以下测试用例:

b---d
-d--d
--dd-
--d--
----d

现在我想5 X 5 Matrix借助python代码保存上面的内容.所以,当我想要返回matrix[0][2]它返回-.

我该怎么做？我尝试了但它打印错误的答案.

Answer 1

这非常简单,因为字符串和文件也是可迭代的:

with open('input.txt') as matrixfile:
    matrix = [list(line.strip()) for line in matrixfile]

list()在一个字符串上将其转换为单个字符的列表; 我们.strip()用来删除任何额外的空格,包括换行符.在matrixfile打开的文件对象是一个迭代的,所以我们可以遍历它来处理所有行.

结果:

>>> matrix
[['b', '-', '-', '-', 'd'], ['-', 'd', '-', '-', 'd'], ['-', '-', 'd', 'd', '-'], ['-', '-', 'd', '-', '-'], ['-', '-', '-', '-', 'd']]
>>> matrix[0][2]
'-'