时间戳

首页 > 解决方案 > 编写一个 Python 函数从输入文件夹中获取 excel 文件，并检查它是 csv 格式还是 xlsx 格式

python - 编写一个 Python 函数从输入文件夹中获取 excel 文件，并检查它是 csv 格式还是 xlsx 格式

问题描述

Python函数从输入文件夹中获取excel文件并检查它是csv格式还是xlsx格式

def import_files(input_file_path):
    for filename in os.listdir(input_file_path):
        if filename.endswith('.xlsx'):
            excel_files=pd.read_excel(input_file_path+'//'+filename,encoding='utf-8',dtype=str)
            print("Read Excel File Name: "+filename)
            print("Displaying Sample Record:")
            print(excel_files.iloc[1])
        else:
            print("File Format Not Supported: "+filename)
    return excel_files

标签： python

解决方案

如果您关心的是获取文件并检查其 excel 或 csv 文件，那么下面是代码：

xlfile=[]
def import_files(input_file_path): 
    for f in os.listdir(input_file_path):
        if f.endswith('.xlsx'):
            xlfile.append(f)
        if f.endswith('.csv'):
            print("File Format Not Supported: ",f)
    return xlfile

excelfiles=import_files(input_file_path)    
print(excelfiles)

推荐阅读