python - 将嵌套列表中的整数输出到文本文件中的字符串

Question

所以我目前在程序末尾有这段代码，允许用户保存文本文件。我不知道如何将数据转换为字符串，如果我尝试使用此代码，这就是给我带来错误的原因。

save = input("Would you like to save the latest generation? ('y' to save):")
if save == 'y':
    destination = input("enter destination file name:")
    with open(destination, 'w') as file:
        file.writelines('\t'.join(i) + '\n' for i in original_graph)
else:
    print("End of program.")

但是，我的 original_graph 是一个仅包含整数值的嵌套列表，例如:[0,1,1,0],[0,0,0,1],[0 ,0,1,0]

我该如何使文本文件看起来像

0110
0001
0010

保存后在文本文件中吗？另外，是否有办法提示用户是否覆盖现有文件？

感谢您的宝贵时间。

Answer 1

一种简洁的编写方法是使用列表推导式。为了方便起见，我将使用 stdout 作为输出文件。

import sys

original_graph = [
    [0, 1, 1, 0],
    [0, 0, 0, 1],
    [0, 0, 1, 0],
]

f = sys.stdout
rows = [''.join([str(u) for u in row]) for row in original_graph]
f.write('\n'.join(rows) + '\n')

输出

0110
0001
0010

要将数据保存到名称为存储在 destination 中的字符串的文件中，只需执行以下操作:

rows = [''.join([str(u) for u in row]) for row in original_graph]
with open(destination, 'w') as f:
    f.write('\n'.join(rows) + '\n')

<小时/>

要检查文件是否已存在，请使用 os.path.exists功能。如果您在这方面需要进一步的帮助，请提出新问题。 Stack Overflow 问题应包含一个问题，以便最大限度地提高对 future 读者的实用性。