python - 如何将 CSV 文件作为参数传递给 python 文件

Question

我正在尝试提供 csv 文件作为 python 文件中的参数，在这里，我想传递 features.csv 文件作为参数。有什么想法如何制作吗？

def read_csv_file():
with open(r'features.csv', 'r') as csvFile:
    checker = lambda i: bool(i and i.strip())
    raw_file = csv.reader(csvFile)
    header = next(raw_file)
    folders = next(
        {
            header[0]: [row[0]],
            'Feature Name': list(filter(checker, row[:1])),
            'Child folder': list(filter(checker, row[1:]))
        } for row in raw_file
    )
    raw_folder_list = list(folders.values())
    folder_list = sum(raw_folder_list, [])
    return folder_list

Answer 1

我认为您正在尝试将文件名作为命令行参数，以便您可以重用此代码来打开所有类似的 csv 文件。

在这种情况下，您可以使用 argparse

或者，简单地说

>> python read_csv.py filename.csv

在read_csv.py中

import sys
first_arg = sys.argv[1]
print(first_arg)
# filename.csv

我建议使用 argparse，因为它更强大。