python遍历包含txt文件的文件夹,将将txt文件的个数从1到最后排序,将序号放在excel文件的第一列,每行的第一个数放在Excel的每行的第二列,若两行或两行以上则放置多个数在第二列以逗号隔开,每个txt的行数放在Excel的第三列,每行的第六个数放在csv的第四列代码,为表格加入表头,第一列表头名为Id,第二列表头为class,第三列表头为number of classes,第四列表头为conf,第五列表头为result的代码,当第二列包含7或8且第三列大于等于2时将第五列标注为1,其余第五列标注为0的代码
时间: 2024-04-10 09:33:08 浏览: 146
以下是用Python遍历包含txt文件的文件夹,将txt文件的个数从1到最后排序,并将序号放在Excel文件的第一列,每行的第一个数放在Excel的每行的第二列,若两行或两行以上则放置多个数在第二列以逗号隔开,每个txt的行数放在Excel的第三列,每行的第六个数放在csv的第四列,并为表格加入表头的代码:
```python
import os
import csv
# 设置文件夹路径和Excel文件名
folder_path = '/path/to/folder'
excel_file = '/path/to/excel.xlsx'
# 遍历文件夹,获取所有txt文件路径
txt_files = []
for root, dirs, files in os.walk(folder_path):
for file in files:
if file.endswith('.txt'):
txt_files.append(os.path.join(root, file))
# 将txt文件按照文件名排序
txt_files.sort()
# 创建Excel文件
with open(excel_file, 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
# 添加表头
writer.writerow(['Id', 'class', 'number of classes', 'conf', 'result'])
# 遍历txt文件并写入Excel
for i, txt_file in enumerate(txt_files, start=1):
with open(txt_file, 'r') as file:
lines = file.readlines()
line_count = len(lines)
# 获取第一个数和第六个数
first_number = int(lines[0].split()[0])
sixth_number = int(lines[0].split()[5])
# 获取第二列的值
second_column_values = [first_number]
for line in lines[1:]:
number = int(line.split()[0])
if number != first_number:
second_column_values.append(number)
# 添加一行数据到Excel
if len(second_column_values) > 1:
second_column_value = ','.join(str(num) for num in second_column_values)
else:
second_column_value = str(first_number)
if (int(second_column_value) in [7, 8]) and line_count >= 2:
result_value = 1
else:
result_value = 0
writer.writerow([i, second_column_value, line_count, sixth_number, result_value])
print("Excel文件生成成功!")
```
请将`/path/to/folder`替换为包含txt文件的文件夹的路径,将`/path/to/excel.xlsx`替换为要保存的Excel文件的路径。执行以上代码后,将会生成一个包含所需数据的Excel文件。
阅读全文