如何通过通用JSON文件向不同Python脚本传递所需参数

我有如下JSON输入参数文件：

# input.json
{
    "nx": 401, 
    "ny": 401,
    "T" : 10,
    "nt" : 20,
    "D" : 0.2,
    "Sgma": 0.243,
    "M": 0.0052
}

该文件需要传递给不同的Python脚本，例如：

# test1.py
import numpy, os

def simple(nx, ny, D, Sgma, M, T, nt):  # 参数作为函数入参
    k = Sgma + 0.02
    print("this is domain size", ny)
    print("this is dif time", T)
    print("this is K param", k)

# test2.py
import numpy, os

def simple_n(nx, ny, D, Sgma, M, T, nt):  # 参数作为函数入参
    k = M + 0.02
    print("this is domain size", ny)
    print("this is sim time", D)
    print("this is K param", k)

我通过main.py调用上述两个脚本，该脚本使用argparse传递输入参数。可以看到每个脚本仅需要部分而非全部参数，我的问题是：能否仅向simple()传递所需的参数？例如test1.py中使用def simple(ny,T,Sgma)，test2.py中使用def simple_n(ny,D,M)，而非必须定义成包含所有参数的函数。由于要处理大量初始配置参数，希望了解在不同Python代码中传递初始参数作为函数参数的高效方法。

1. 仅传递所需参数：完全可行 #

你可以直接修改函数定义，只保留实际需要的参数（注意：k是函数内部计算得到的，不需要作为入参传入，之前的示例里写def simple(ny,T,k)是误区，应该传入用来计算k的原始参数，比如Sgma或M）。

修改后的test1.py示例：

# test1.py
import numpy, os

def simple(ny, T, Sgma):  # 只保留函数实际需要的参数
    k = Sgma + 0.02
    print("this is domain size", ny)
    print("this is dif time", T)
    print("this is K param", k)

修改后的test2.py示例：

# test2.py
import numpy, os

def simple_n(ny, D, M):  # 只保留函数实际需要的参数
    k = M + 0.02
    print("this is domain size", ny)
    print("this is sim time", D)
    print("this is K param", k)

2. 高效传递参数的实用方法 #

方法一：解析JSON为字典，按需提取参数

在main.py中先把JSON文件解析成字典，然后根据每个函数的需求提取对应键值，通过关键字参数传递：

# main.py
import argparse
import json
from test1 import simple
from test2 import simple_n

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--input', type=str, required=True, help='输入JSON文件路径')
    args = parser.parse_args()

    # 解析JSON文件为字典
    with open(args.input, 'r') as f:
        params = json.load(f)

    # 给test1的simple传递所需参数
    simple(ny=params['ny'], T=params['T'], Sgma=params['Sgma'])

    # 给test2的simple_n传递所需参数
    simple_n(ny=params['ny'], D=params['D'], M=params['M'])

if __name__ == '__main__':
    main()

这种方式逻辑清晰，每个函数只接收自己需要的参数，避免冗余。

方法二：关键字参数解包（适合参数名与JSON键名一致的场景）

如果函数参数名和JSON中的键名完全匹配，可以通过字典推导筛选出需要的键，再用**解包传递，代码更简洁：

# main.py 中调用函数的部分
# 给test1传参
required_test1 = ['ny', 'T', 'Sgma']
test1_params = {key: params[key] for key in required_test1}
simple(**test1_params)

# 给test2传参
required_test2 = ['ny', 'D', 'M']
test2_params = {key: params[key] for key in required_test2}
simple_n(**test2_params)

方法三：用dataclass封装参数（适合大量参数的场景）

如果参数数量极多，且需要类型提示和结构化管理，可以用dataclasses模块封装参数，后续按需提取属性传递：

# main.py
import argparse
import json
from dataclasses import dataclass
from test1 import simple
from test2 import simple_n

@dataclass
class Config:
    nx: int
    ny: int
    T: float
    nt: int
    D: float
    Sgma: float
    M: float

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--input', type=str, required=True, help='输入JSON文件路径')
    args = parser.parse_args()

    # 解析JSON为Config实例
    with open(args.input, 'r') as f:
        config = Config(**json.load(f))

    # 传递参数
    simple(ny=config.ny, T=config.T, Sgma=config.Sgma)
    simple_n(ny=config.ny, D=config.D, M=config.M)

if __name__ == '__main__':
    main()

这种方式自带类型检查和代码提示，维护大型项目时更高效。

3. 注意事项 #

• 函数内部可推导的参数（比如k）不要作为入参传递，避免参数冗余。
• 尽量使用关键字参数传递，避免因位置顺序出错导致的bug。
• 若多个函数共享部分参数，可以提取公共参数集合，减少重复代码。

内容的提问来源于stack exchange，提问作者newstudent