写一个python代码,将一个excel的前三列数据写入文件夹内所有excel的前三列
时间: 2024-05-03 21:23:24 浏览: 10
以下是一个可行的Python代码示例:
```python
import os
import pandas as pd
# 定义excel文件夹路径
folder_path = "path/to/your/excel/folder"
# 获取文件夹内所有excel文件的路径
file_paths = [os.path.join(folder_path, file_name) for file_name in os.listdir(folder_path) if file_name.endswith('.xlsx')]
# 遍历每个excel文件,读取前三列数据并写入新文件
for file_path in file_paths:
df = pd.read_excel(file_path, usecols=[0, 1, 2]) # 读取前三列数据
new_file_path = os.path.splitext(file_path)[0] + "_new.xlsx" # 定义新文件路径
df.to_excel(new_file_path, index=False) # 写入新文件,不写入行索引
```
此代码首先定义了一个excel文件夹的路径,然后获取该文件夹内所有以`.xlsx`为后缀的文件路径。接着,代码遍历每个excel文件,使用pandas库的`read_excel()`方法读取前三列数据,并使用`os.path.splitext()`方法生成新文件的路径。最后,使用`to_excel()`方法将前三列数据写入新文件。
需要注意的是,这个代码假设每个excel文件都有至少三列数据。如果有些文件只有两列或更少的数据,代码将引发异常。如需解决此问题,可以在读取数据时使用try-except块,以便在读取失败时将其跳过。