python中如何判断文件是否为视频格式,代码
时间: 2024-03-14 08:47:21 浏览: 193
Python推荐视频网站代码.zip
代码如下所示:
```python
import os
# 定义视频格式的扩展名列表
VIDEO_EXTENSIONS = ['.mp4', '.avi', '.mov', '.wmv', '.flv']
# 判断文件是否为视频格式
def is_video_file(file_path):
# 获取文件扩展名并转换为小写
ext = os.path.splitext(file_path)[-1].lower()
# 判断文件扩展名是否在视频格式列表中
return ext in VIDEO_EXTENSIONS
# 测试代码
file_path = 'test.mp4'
if is_video_file(file_path):
print(f'{file_path} 是视频文件')
else:
print(f'{file_path} 不是视频文件')
```
这段代码中,我们首先定义了一个视频格式的扩展名列表 `VIDEO_EXTENSIONS`,然后定义了一个函数 `is_video_file`,该函数接受一个文件的路径作为参数,判断该文件的扩展名是否在视频格式列表中,如果在则返回 `True`,否则返回 `False`。最后我们使用测试代码来调用该函数,测试文件是否为视频文件。
阅读全文