Python新手求助：如何在脚本中用变量值加载指定文件

嘿，作为Python初学者能想到用命令行参数来动态指定路径，这个思路真的不错！我来一步步帮你实现这个需求，让程序能根据传入的folder_name（比如abc）正确加载目标文件～

1. 选择合适的方式获取命令行参数 #

Python里获取命令行参数有两种常用方式：简单直接的sys模块，或者更规范灵活的argparse模块。我都给你讲讲，你可以按需选择。

方式一：用sys模块快速获取 #

sys.argv是一个列表，里面保存了命令行输入的所有内容，sys.argv[0]是脚本本身的名字，后面的元素就是你传入的参数。代码示例：

import sys

# 先检查用户有没有传入参数，避免报错
if len(sys.argv) < 2:
    print("请记得传入文件夹名称哦！比如：python your_script.py abc")
    sys.exit(1)  # 没有参数就退出程序

# 把第一个参数赋值给folder_name
folder_name = sys.argv[1]

方式二：用argparse模块（推荐！） #

这个模块能帮你定义参数说明，自动生成帮助信息，更适合复杂一点的场景，也更易读。代码示例：

import argparse

# 创建一个参数解析器，添加描述信息
parser = argparse.ArgumentParser(description='根据指定文件夹加载目标文件')
# 添加folder_name参数，说明它是字符串类型，还有帮助提示
parser.add_argument('folder_name', type=str, help='目标文件夹的名称，比如abc')

# 解析用户传入的参数
args = parser.parse_args()
# 把参数值赋值给变量
folder_name = args.folder_name

运行的时候如果输入python your_script.py -h，会自动显示帮助信息，非常友好～

2. 拼接路径并加载文件 #

拿到folder_name之后，接下来要拼接出文件的完整路径。这里强烈推荐用os.path.join()来拼接，它能自动适配Windows和Linux的路径分隔符（\和/），避免跨平台问题。

假设你要加载的文件是data.txt，放在目标文件夹里，代码如下：

import os

# 拼接完整文件路径
file_path = os.path.join(folder_name, 'data.txt')

# 先检查文件是否存在，避免读取时报错
if not os.path.exists(file_path):
    print(f"哎呀，找不到文件：{file_path}")
    sys.exit(1)

# 读取文件内容，赋值给变量
with open(file_path, 'r', encoding='utf-8') as f:
    file_content = f.read()

# 可以打印看看结果
print(f"成功加载文件！内容如下：\n{file_content}")

3. 完整可运行示例 #

把上面的代码整合起来，用argparse的完整版本，还加了异常处理，更健壮：

import argparse
import os

def load_target_file(folder_name):
    # 拼接文件路径
    target_file_path = os.path.join(folder_name, 'data.txt')
    # 检查文件是否存在且是文件（不是文件夹）
    if not os.path.isfile(target_file_path):
        raise FileNotFoundError(f"目标文件不存在：{target_file_path}")
    # 读取文件内容
    with open(target_file_path, 'r', encoding='utf-8') as f:
        return f.read()

if __name__ == '__main__':
    # 处理命令行参数
    parser = argparse.ArgumentParser(description='从指定文件夹加载文件')
    parser.add_argument('folder_name', type=str, help='文件夹名称，例如abc')
    args = parser.parse_args()
    
    try:
        # 调用加载函数
        content = load_target_file(args.folder_name)
        print("文件加载成功！内容：")
        print(content)
    except Exception as e:
        print(f"运行出错：{e}")

怎么测试？ #

假设你的脚本名叫load_file.py，当你在命令行输入：

python load_file.py abc

程序就会去abc/data.txt路径下读取文件，把内容赋值给content变量，然后打印出来。如果folder_name是abc，就会自动用这个路径执行啦～

额外小提示 #

• 如果你的目标文件不是txt，比如json、csv，只需要调整读取方式：比如用json.load(f)读取json文件，用csv.reader(f)读取csv文件。
• 可以给folder_name设置默认值，比如在argparse里改成：

  parser.add_argument('folder_name', type=str, help='文件夹名称', default='abc', nargs='?')
  

  这样即使不传参数，程序也会默认用abc作为文件夹名称～

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