Python如何通过命令行参数传入路径加载数据集至pandas DataFrame

你可以用Python内置的argparse标准库实现你要的-D指定参数的效果，不需要额外安装第三方依赖，完整实现步骤如下：

步骤1：编写example.py代码 #

import pandas as pd
import argparse

# 初始化参数解析器
parser = argparse.ArgumentParser(description='加载指定路径的csv数据集到DataFrame')
# 添加-D参数，用于接收数据集路径
parser.add_argument('-D', '--dataset', required=True, type=str, help='csv数据集的路径')
# 解析命令行传入的参数
args = parser.parse_args()

# 读取数据集
df = pd.read_csv(args.dataset, header=0)
# 测试输出前5行
print(df.head())

步骤2：执行命令运行脚本 #

直接运行你预期的命令即可：

python3 example.py -D 'iris_dataset.csv'

如果你的数据集路径包含空格，只需要把路径放在单引号/双引号里即可正常识别。

轻量替代方案（不需要参数标识） #

如果不需要-D参数标识，只是简单按位置传参，可以用更轻量化的sys库实现：

import pandas as pd
import sys

# 取命令行传入的第一个位置参数作为数据集路径
dataset_path = sys.argv[1]
df = pd.read_csv(dataset_path, header=0)
print(df.head())

运行命令为：

python3 example.py 'iris_dataset.csv'

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