def zipjie(): f = zipfile.ZipFile("E:\daad\daaad\Ad.rar", 'r') # 压缩文件位置 for file in f.namelist(): f.extract(file, "E:\daad\daaad") # 解压位置 f.close() # 指定要查找的文件夹路径 folder_path = "E:\daad\daaad" # 指定要选取的字段 selected_columns = ['dsada', 'dasda'] # 使用 os 模块列出文件夹中所有的 XLS 文件 xls_files = [f for f in os.listdir(folder_path) if f.endswith('.xls')] # 检查 Excel 文件是否存在 writer = pd.ExcelWriter('E:\daad\daaad\qwd.xlsx') # 循环对每个 XLS 文件进行操作 for xls_file in xls_files: # 构造文件路径 file_path = os.path.join(folder_path, xls_file) # 使用 pandas 打开 XLS 文件 df = pd.read_excel(file_path) # 选取指定的字段,并判断,若不存在的字段跳过 valid_columns = [col for col in selected_columns if col in df.columns] if len(valid_columns) < len(selected_columns): print(f"Warning: {xls_file} contains invalid column names, skipping them") # 去除特殊符号和非法字符 for col in valid_columns: df[col] = df[col].apply(lambda x: re.sub(r'[^\w\s]', '', str(x))) df[col] = df[col].apply(lambda x: re.sub(r'[\d]', '', str(x))) selected_data = df[valid_columns] # 将选取的数据追加到现有的 Excel 文件中 selected_data.to_excel(writer, sheet_name=xls_file, index=False, header=False, startrow=len(pd.read_excel(file_path, sheet_name=xls_file))) # 保存并关闭新的 Excel 文件 writer.save() writer.close() zipjie() 帮我改造下代码,实现计算已有EXCEL的行数,然后对EXCEL内容进行追加
时间: 2024-02-14 09:13:05 浏览: 120
ZipFile.dll 压缩解压第三方类库
你可以使用 Pandas 库中的 `pd.read_excel()` 方法读取已有 Excel 文件,并使用 `shape[0]` 方法获取文件的行数,然后在追加数据时指定 `startrow` 参数为行数即可。下面是改造后的代码:
```python
import os
import re
import pandas as pd
import zipfile
def unzip_and_append():
# 解压缩文件
with zipfile.ZipFile("E:\daad\daaad\Ad.rar", 'r') as f:
f.extractall("E:\daad\daaad")
# 指定要查找的文件夹路径
folder_path = "E:\daad\daaad"
# 指定要选取的字段
selected_columns = ['dsada', 'dasda']
# 使用 os 模块列出文件夹中所有的 XLS 文件
xls_files = [f for f in os.listdir(folder_path) if f.endswith('.xls')]
# 检查 Excel 文件是否存在
writer = pd.ExcelWriter('E:\daad\daaad\qwd.xlsx', engine='xlsxwriter')
for xls_file in xls_files:
# 构造文件路径
file_path = os.path.join(folder_path, xls_file)
# 获取已有 Excel 文件的行数
try:
startrow = pd.read_excel(writer, sheet_name=xls_file).shape[0] + 1
except:
startrow = 0
# 使用 pandas 打开 XLS 文件
df = pd.read_excel(file_path)
# 选取指定的字段,并判断,若不存在的字段跳过
valid_columns = [col for col in selected_columns if col in df.columns]
if len(valid_columns) < len(selected_columns):
print(f"Warning: {xls_file} contains invalid column names, skipping them")
# 去除特殊符号和非法字符
for col in valid_columns:
df[col] = df[col].apply(lambda x: re.sub(r'[^\w\s]', '', str(x)))
df[col] = df[col].apply(lambda x: re.sub(r'[\d]', '', str(x)))
selected_data = df[valid_columns]
# 将选取的数据追加到现有的 Excel 文件中
selected_data.to_excel(writer, sheet_name=xls_file, index=False, header=False,
startrow=startrow)
# 保存并关闭新的 Excel 文件
writer.save()
writer.close()
unzip_and_append()
```
这样,代码就可以计算已有 Excel 文件的行数,并将新的数据追加到文件中。
阅读全文