python遍历包含txt文件的文件夹,将将txt文件的个数从1到最后排序,将序号放在csv文件的第一列,每行的第一个数放在csv的每行的第二列,若两行或两行以上则放置多个数在第二列以逗号隔开,每个txt的行数放在csv的第三列,每行的第六个数放在csv的第四列代码,为表格加入表头,第一列表头名为Id,第二列表头为class,第三列表头为number of classes,第四列表头为conf,第五列表头为result的代码,当第二列包含7或8且第三列大于等于2时将第五列标注为1,其余第五列标注为0的代码
时间: 2024-04-10 17:33:37 浏览: 76
以下是根据您的要求编写的Python代码,用于遍历包含txt文件的文件夹并将数据写入CSV文件:
```python
import os
import csv
# 定义文件夹路径和CSV文件路径
folder_path = 'your_folder_path' # 替换为包含txt文件的文件夹路径
csv_file = 'output.csv' # 替换为输出的CSV文件路径
# 创建一个CSV文件并打开
with open(csv_file, 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
# 写入表头
writer.writerow(['Id', 'class', 'number of classes', 'conf', 'result'])
# 遍历文件夹中的txt文件并按文件名排序
txt_files = sorted([f for f in os.listdir(folder_path) if f.endswith('.txt')])
for idx, file_name in enumerate(txt_files, start=1):
file_path = os.path.join(folder_path, file_name)
with open(file_path, 'r') as file:
lines = file.readlines()
col2_values = [float(line.split()[0]) for line in lines] # 获取每行的第一个数并转换为浮点型
col2 = ', '.join(str(val) for val in col2_values) # 多个数以逗号隔开
col3 = len(lines) # 获取行数
col6 = float(lines[0].split()[5]) # 获取每行的第六个数并转换为浮点型
# 判断条件以设置result的值
if '7' in file_name or '8' in file_name:
if col3 >= 2:
col5 = 1
else:
col5 = 0
else:
col5 = 0
# 将数据写入CSV文件的相应位置
writer.writerow([idx, file_name, col3, col6, col5])
```
请将`your_folder_path`替换为包含txt文件的文件夹路径,并将`output.csv`替换为要输出的CSV文件路径。上述代码会遍历文件夹中的txt文件,并按照文件名进行排序。每个文件的序号将放在CSV文件的第一列,每行的第一个数将以逗号隔开放在第二列,行数放在第三列,每行的第六个数放在第四列。CSV文件的表头是"Id"、"class"、"number of classes"、"conf"和"result"。根据您的要求,当第二列包含7或8且第三列大于等于2时,第五列标注为1,否则标注为0。
阅读全文