帮我用python写一个从50个txt文本中提取第五行和第7行的数据,并让这两组数据一一对应且用csv库生成一个excel表格,并把表格存放在D盘根目录
时间: 2024-05-02 09:19:53 浏览: 17
好的,以下是Python代码实现:
```python
import csv
import os
# 设置文本文件夹路径
folder_path = 'C:/txt_files'
# 创建一个空列表,用于存放所有文件的第五行和第七行数据
data_list = []
# 遍历文件夹中的所有txt文件
for filename in os.listdir(folder_path):
if filename.endswith('.txt'):
file_path = os.path.join(folder_path, filename)
with open(file_path, 'r', encoding='utf-8') as f:
lines = f.readlines()
# 获取第五行和第七行的数据
data_5 = lines[4].strip()
data_7 = lines[6].strip()
# 将两组数据存入列表中
data_list.append((data_5, data_7))
# 写入csv文件
csv_path = 'D:/data.csv'
with open(csv_path, 'w', newline='', encoding='utf-8') as f:
writer = csv.writer(f)
# 写入表头
writer.writerow(['data_5', 'data_7'])
# 遍历数据列表,逐行写入csv文件
for data in data_list:
writer.writerow(data)
print('数据已成功提取并存储至D盘根目录下的data.csv文件!')
```
请确保修改 `folder_path` 和 `csv_path` 变量,使其符合你的本地文件路径。如果你需要从其他文件夹中提取数据,请将 `folder_path` 变量修改为正确的文件夹路径。
希望这个代码能够帮到你,如有任何问题,请随时回复!