从其他/程序python调用函数

Question

`第一个程序：first.py

list=["ab","cd","ef"]
for i in list:
    with open("input.txt", "w") as input_file:
        print(" {}".format(i), file = input_file)

预期输出：

ab
cd
ef

但我得到了输出：

ef

第二个程序：second.py

input_file = open('input.txt','r')     

for line in input_file:
    if "ef" in line:
       print(line)

预期输出：

ef

得到输出：

ef

现在我想直接从 first.py 调用文本文件（input.txt）并在 second.py 中使用它？`如何从其他程序 python 调用函数？

编辑：应用的代码块

Answer 1

您在for 循环中打开文件，并使用w 作为open 函数的模式参数，它使open 覆盖它打开的文件，这就是为什么您只能得到循环最后一次迭代的输出。

您应该改为在循环外打开文件：

with open("input.txt", "w") as input_file:
    for i in list:
        print("{}".format(i), file = input_file)

Answer 2

在first.py 中，像这样更改您的代码。

w 模式用于写操作。在 for 循环的每次迭代中，您都将覆盖最后一个内容并编写新内容。所以input.txt 里面有ef（最后）。

list=["ab","cd","ef"]

for i in list:
    with open("input.txt", "a+") as input_file:
        print("{}".format(i), file = input_file)

现在你会得到你所期望的。现在input.txt 将与您的情况不同。

ab
cd
ef

注意：但如果您将第二次运行first.py，它将继续添加，因为a+ 如果文件不存在则创建文件，否则它会追加。 为了更好地运行此代码，请使用 os.path 模块的 exists() 函数。

如果您想调用first.py 中可用的代码，请将其包装在函数中。然后将该函数导入second.py 并调用。

例如

首先确保first.py 和second.py 在同一个目录中。

first.py

def create_file(file_name):
    list=["ab","cd","ef"]
    for i in list:
        with open(file_name, "a+") as input_file:
            print(" {}".format(i), file = input_file)

second.py

from first import create_file

def read_file(file_name):
    # Create file with content
    create_file(file_name)

    # Now read file content
    input_file = open(file_name, 'r')     
    for line in input_file:
        if "ef" in line:
           print(line)

read_file('input.txt')

打开终端，导航到这个目录，运行python second.py。

https://www.learnpython.org/en/Module... | https://www.digitalocean.com... |如果您想阅读并尝试如何在 Python 中创建模块/包，https://www.programiz.com/pytho... 将是您的帮手。

更新：上面有你在评论中提到的问题，在每次运行时，它都会附加内容。让我们通过对first.py 进行一些更改来修复它，如下所示。

import os

def create_file(file_name):
    l = ["ab", "cd", "ef"]

    if os.path.exists(file_name): # If file `input.txt` exists (for this example)
        os.remove(file_name)      # Delete the file

    for i in l:
        with open(file_name, "a+") as input_file:
            print(" {}".format(i), file = input_file)

就是这样（如果您遇到困难，请在评论中更新）。