先试用主函数实现,随后使用自定义函数实现,将一组整数写到文件中(数组到磁盘文件
时间: 2023-12-26 20:03:26 浏览: 32
好的,以下是主函数实现:
```python
def main():
# 定义一组整数
nums = [1, 2, 3, 4, 5]
# 打开文件并写入整数
with open('nums.txt', 'w') as f:
for num in nums:
f.write(str(num) + '\n')
# 提示写入成功
print('写入成功!')
if __name__ == '__main__':
main()
```
这里我们使用了 `with` 语句来打开文件并写入整数。该语句会自动关闭文件,无需手动调用 `f.close()` 函数。
接下来是自定义函数实现:
```python
def write_nums(nums, filename):
# 打开文件并写入整数
with open(filename, 'w') as f:
for num in nums:
f.write(str(num) + '\n')
# 提示写入成功
print('写入成功!')
```
这里我们把之前的代码封装成了一个名为 `write_nums` 的函数,其中 `nums` 是需要写入的整数数组,`filename` 是要写入的文件名。调用方式如下:
```python
nums = [1, 2, 3, 4, 5]
filename = 'nums.txt'
write_nums(nums, filename)
```