python中argparse库,在B文件中如何调取A文件中的args参数,直接调用的方式

如果你想在 B 文件中导入 A 文件并直接访问其中的参数,那么需要注意以下两点:

  1. A 文件中定义的参数必须是全局变量或者是可被外部访问的类属性或实例属性。否则,在 B 文件中无法直接访问这些参数。

  2. 如果 A 文件中定义了和 B 文件中相同名称的参数,那么在导入 A 文件之后,B 文件中的同名参数将被覆盖,导致访问到的值可能不是预期的值。

假设我们有两个 Python 文件 A.pyB.py,其中 A.py 中定义了一些命令行参数并对其进行了解析,现在我们想要在 B.py 文件中导入 A.py 并直接访问其中的参数。可以按照以下步骤进行操作:

A 文件中,定义一个函数或类方法,用于返回解析得到的参数对象。示例代码如下:

import argparse
# 创建 ArgumentParser 对象
parser = argparse.ArgumentParser()
# 添加要解析的命令行参数
parser.add_argument('--lr', type=float, default=0.01)
parser.add_argument('--batch_size', type=int, default=32)
def get_args():
    # 解析命令行参数并返回参数对象
    return parser.parse_args()

在上述代码中,我们定义了一个名为 get_args() 的函数,该函数使用 ArgumentParser 解析命令行参数,并返回解析得到的参数对象。

B 文件中,导入 A 文件并调用 get_args() 函数来获取解析得到的参数对象。示例代码如下:

import A  # 导入 A 文件
# 调用 get_args() 函数获取参数对象
args = A.get_args()
# 打印读取的参数
print(args.lr)
print(args.batch_size)

在上述代码中,我们首先导入了 A 文件,并使用 A.get_args() 函数获取解析得到的参数对象。然后,我们可以直接访问该对象中的属性,如 args.lrargs.batch_size

需要注意的是,在实际应用中,如果多个 Python 文件需要共享相同的参数值,建议使用配置文件或环境变量等方式来存储和传递参数值,而不是直接在 Python 文件中定义和访问全局变量或类属性。这样可以提高代码的可维护性和灵活性。

猜你喜欢

转载自blog.csdn.net/AdamCY888/article/details/134750045