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

Question

So I currently have this code at the end of my program that allows the user to save the text file. I have no idea how to convert the data to string and that is what is giving me the error if i try to use this code.

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.")

However, my original_graph is a nested list with only integer values, ex: [0,1,1,0],[0,0,0,1],[0,0,1,0]

How would I make it so that the text file is looks something like

0110 0001 0010

in the text file after I save it? Also, is there a way to prompt the user if they were overwrite an existing file?

Thank you for your time.

Answer 1

A compact way to write this is using list comprehensions. For convenience, I'll use stdout as the output file.

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')

output

0110
0001
0010

To save the data to a file whose name is a string stored in destination , just do:

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')

---

To check if a file already exists, use the os.path.exists function. If you need further help on this aspect please ask a new question. Stack Overflow questions should contain a single question in order to maximize their usefulness for future readers.

Answer 2

Please check this works

original_graph=[[0,1,1,0],[0,0,0,1],[0,0,1,0]]
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(str(i) for i in original_graph))
else:
    print("End of program.")