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

Question

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

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

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

现在我想借助python代码将以上内容保存在5 X 5 Matrix中。这样当我想返回 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]
'-'