写代码以迭代目录中的所有文件，并将每个文件的词典路径附加

Question

我需要在Python模块中实现以下内容。

1. 通过ARGPARSE模块使用该目录作为参数，
2. 该目录在该目录中的所有文件中迭代目录。
3. 后来，仅将CSV文件附加到字典中并将其存储在此处。

我尝试了操作系统，但它没有返回任何东西。

我已经编写了代码以获取目录作为输入，然后迭代文件以获取文件。但这会引发错误。

以下是代码

 import argparse
 import os
 import re
    
 # Accepting arguments for source_directory, my_sql connection details and table name to 
   update the MySQL entries to
    
parser = argparse.ArgumentParser(
        description='Utility to update MySQL tables on remote host')
parser.add_argument('-s', '--source_dir', help='Enter Path For CSV File')
    
args = parser.parse_args()
    
# Empty dictionary to store list CSV files in directory
files_in_dir = {}
    
# Function to iterate through the source_directory and read-only CSV files and append them 
   to files_in_dir dictionary
    
if args.source_dir:
    for file in args.source_dir:
    
        if os.path.abspath(file.name).endswith('.csv'):
            files_in_dir[(os.path.abspath(file.name))] = 0
    
        else:
            print("The file %s is not a .csv file and it will be ignored" % (file.name))
    
print(files_in_dir)

，这是通过终端执行时的错误。

 python3 sourcecheck.py -s /home/thepredator/Desktop/VOIS/csv_files/*
usage: sourcecheck.py [-h] [-s SOURCE_DIR]
sourcecheck.py: error: unrecognized arguments: /home/thepredator/Desktop/VOIS/csv_files/Book1.csv /home/thepredator/Desktop/VOIS/csv_files/Build.py /home/thepredator/Desktop/VOIS/csv_files/car_data.csv /home/thepredator/Desktop/VOIS/csv_files/db.py /home/thepredator/Desktop/VOIS/csv_files/Edited.csv /home/thepredator/Desktop/VOIS/csv_files/Iris.csv /home/thepredator/Desktop/VOIS/csv_files/Lab_Python_for_Data_Analytics.ipynb /home/thepredator/Desktop/VOIS/csv_files/startupNaN.csv /home/thepredator/Desktop/VOIS/csv_files/startupog.csv /home/thepredator/Desktop/VOIS/csv_files/text.csv

我通过观看不同的视频来编写上述代码。请帮助我解决错误。另外，我不是Python专家。

Answer 1

我更改了files_dir = {} line

files_in_dir = { key:0 for key in os.listdir(args.source_dir) if os.path.abspath(key).endswith(".csv") }

and it seems work.

for语句版本以下请求：

for key in os.listdir(args.source_dir):
    if os.path.abspath(key).endswith(".csv"):
        files_in_dir[key] = 0
    else:
        print(f"The file {key} is not a .csv file and it will be ignored")

Answer 2

您可以使用此代码`

import os

files = os.listdir("/path_to_directory")    

csv = list(filter(lambda f: f.endswith('.csv'), files))

csv具有.csv扩展名的所有文件