在Python中,我们可以使用os模块的path.splitext()函数来获取文件扩展名,然后判断这个扩展名是否为".csv"。以下是一个示例代码:
import os
def is_csv_file(file_path):
_, file_extension = os.path.splitext(file_path)
return file_extension == '.csv'
# 测试代码
file_path = "example.csv"
if is_csv_file(file_path):
print(f"{file_path} 是一个CSV文件")
else:
print(f"{file_path} 不是一个CSV文件")
这个代码中的`is_csv_file`函数接受一个文件路径作为参数,然后通过`os.path.splitext()`函数把文件路径分成两部分,第一部分是文件名,第二部分是文件的扩展名。然后它比较文件的扩展名是否是".csv"。如果是,那么这个文件就是一个CSV文件,函数返回True;否则,这个文件不是一个CSV文件,函数返回False。
需要注意的是,这个代码只检查文件的扩展名,而不会打开文件本身来检查其内容。所以如果一个文件被错误地命名为一个CSV文件的扩展名,这个代码也会认为它是CSV文件。如果你需要更精确的检测,可能需要打开文件并读取其内容。