读取一个文件夹下所有csv文件中特定行的信息,并且写入到一个excel中
时间: 2023-05-29 11:03:09 浏览: 144
这里提供一个Python的代码示例,使用pandas库来读取csv文件,并且使用openpyxl库来写入excel文件。假设我们需要读取文件夹"data"下所有csv文件中的第3行和第5行,并且将这些信息写入到一个excel文件"output.xlsx"中。
首先需要安装依赖库:
```
pip install pandas openpyxl
```
代码示例:
```python
import os
import pandas as pd
from openpyxl import Workbook
# 文件夹路径
folder_path = "data"
# 读取csv文件中的特定行
def read_csv(file_path):
df = pd.read_csv(file_path)
row3 = df.iloc[2].tolist() # 第3行转换成列表
row5 = df.iloc[4].tolist() # 第5行转换成列表
return row3 + row5 # 合并成一个列表
# 遍历文件夹下的所有csv文件
csv_files = [f for f in os.listdir(folder_path) if f.endswith(".csv")]
# 创建excel文件
wb = Workbook()
ws = wb.active
# 写入第一行的表头
ws.append(["文件名", "第3行", "第5行"])
# 读取csv文件并写入excel
for file in csv_files:
file_path = os.path.join(folder_path, file)
data = read_csv(file_path)
ws.append([file] + data)
# 保存excel文件
wb.save("output.xlsx")
```
上述代码会在当前目录下创建一个名为"output.xlsx"的excel文件,并且将所有csv文件中的第3行和第5行信息写入到该文件中。其中,第一行为表头,包括"文件名"、"第3行"和"第5行"三列。每个csv文件对应一行数据,包括文件名和该文件第3行和第5行的信息。
阅读全文