Python 在Python脚本之间传递变量

在本文中，我们将介绍如何在Python脚本之间传递变量。在实际开发中，经常会遇到需要在不同的Python脚本之间共享数据的情况。了解如何传递变量将极大地提高我们的代码灵活性和可扩展性。

方法1：使用命令行参数

一种简单的方法是使用命令行参数来传递变量。在调用脚本时，可以传递参数给Python脚本，脚本可以通过sys.argv获取这些参数。下面是一个示例：

# script1.py
import sys

def main():
    # 获取传递的参数
    variable = sys.argv[1]
    print("传递的变量是：", variable)

if __name__ == "__main__":
    main()

在命令行中执行以下命令：

python script1.py Hello

输出将是：

传递的变量是： Hello

在脚本中，可以通过sys.argv[1]访问到传递的变量。需要注意的是，sys.argv列表中的第一个元素是脚本的名称，第二个元素才是第一个传递的参数。

方法2：使用环境变量

另一种常见的传递变量的方法是使用环境变量。Python提供了os模块，可以用于获取和设置环境变量。下面是一个示例：

# script1.py
import os

def main():
    # 获取环境变量的值
    variable = os.getenv("MY_VARIABLE")
    print("传递的变量是：", variable)

if __name__ == "__main__":
    main()

在命令行中执行以下命令：

export MY_VARIABLE=Hello
python script1.py

输出将是：

传递的变量是： Hello

在脚本中，os.getenv()函数可以获取指定环境变量的值。在命令行中，可以使用export命令设置环境变量的值。

方法3：使用文件

另一种传递变量的方法是使用文件。可以将变量的值写入文件，在另一个脚本中读取该文件。下面是一个示例：

# script1.py
def main():
    # 将变量写入文件
    with open("variable.txt", "w") as file:
        file.write("Hello")

if __name__ == "__main__":
    main()

# script2.py
def main():
    # 从文件中读取变量的值
    with open("variable.txt", "r") as file:
        variable = file.read()
    print("传递的变量是：", variable)

if __name__ == "__main__":
    main()

首先，在命令行中执行以下命令：

python script1.py

然后执行以下命令：

python script2.py

输出将是：

传递的变量是： Hello

在第一个脚本中，使用with open()语句将变量的值写入文件。在第二个脚本中，同样使用with open()语句从文件中读取变量的值。

方法4：使用模块

还有一种更高级的方法是使用自定义模块来传递变量。可以将共享的变量定义在一个模块中，并在其他脚本中导入该模块。这样，所有导入该模块的脚本都可以访问共享的变量。下面是一个示例：

# shared_variable.py
shared_variable = "Hello"

# script1.py
import shared_variable

def main():
    # 使用共享的变量
    print("传递的变量是：", shared_variable.shared_variable)

if __name__ == "__main__":
    main()

# script2.py
import shared_variable

def main():
    # 使用共享的变量
    print("传递的变量是：", shared_variable.shared_variable)

if __name__ == "__main__":
    main()

在上述示例中，将共享的变量定义在一个名为shared_variable.py的模块中。然后，在其他脚本中通过import语句导入该模块，并使用shared_variable.shared_variable访问该变量。