数组中区分pdf还是image或者两者都存在
时间: 2023-11-12 19:06:16 浏览: 64
Python 实现将数组/矩阵转换成Image类
要区分数组中的文件是PDF还是图像,可以检查文件的扩展名。通常,PDF文件的扩展名为“.pdf”,而图像文件的扩展名可以是“.jpg”、“.png”、“.gif”等。
可以使用以下方法来检查数组中的每个文件的扩展名:
1. 使用循环遍历数组中的每个文件;
2. 对于每个文件,使用字符串函数获取其扩展名;
3. 如果扩展名为“.pdf”,则将该文件添加到PDF文件数组中;
4. 如果扩展名为“.jpg”、“.png”、“.gif”等,则将该文件添加到图像文件数组中;
5. 如果扩展名不是以上任何一种,则可以将其视为未知类型的文件。
下面是一个示例代码,用于将一个文件数组分成PDF和图像两个数组:
```python
pdf_files = []
image_files = []
for file in file_array:
file_ext = file.split('.')[-1].lower()
if file_ext == 'pdf':
pdf_files.append(file)
elif file_ext in ['jpg', 'jpeg', 'png', 'gif']:
image_files.append(file)
else:
print('Unknown file type: ' + file)
```
这个代码将会把文件数组中的文件按照扩展名分成两个数组,其中一个包含PDF文件,另一个包含图像文件。
阅读全文