我想要这个代码同时可以处理xlsx文件和xls文件和xlt文件,应该怎么修改代码,请发给我一份修改好的代码
时间: 2024-12-04 15:29:32 浏览: 11
为了处理Excel文件(`.xls`, `.xlsx`, 和 `.xlt`),你可以使用Python的pandas库,它支持多种Excel文件格式。这里是一个基本示例,展示如何使用pandas统一读取这三种类型的文件:
```python
import pandas as pd
def read_excel(file_path):
try:
# 使用try-except块处理不同格式的文件
if file_path.endswith('.xls'): # Excel 97-2004 (.xls)
df = pd.read_excel(file_path, engine='xlrd')
elif file_path.endswith(('.xlsx', '.xlt')): # Excel 2007+ (.xlsx)
df = pd.read_excel(file_path, engine='openpyxl')
else:
raise ValueError("Unsupported file type. Please provide an .xls, .xlsx, or .xlt file.")
# 如果读取成功,返回DataFrame
return df
except FileNotFoundError:
print(f"File {file_path} not found.")
except Exception as e:
print(f"Error reading file: {str(e)}")
# 调用函数并处理结果
file_to_read = 'example.xlsx' # 替换为你需要处理的文件路径
data = read_excel(file_to_read)
if data is not None:
# 这里你可以对数据进行操作
print(data.head()) # 打印前几行数据
阅读全文