86 lines
3.4 KiB
Python
86 lines
3.4 KiB
Python
|
import os
|
|||
|
|
import stat
|
|||
|
|
import math
|
|||
|
|
|
|||
|
|
|
|||
|
|
def make_writable(file_path):
|
|||
|
|
os.chmod(file_path, stat.S_IWRITE)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def process_files_in_folder(folder_path):
|
|||
|
|
for root, _, files in os.walk(folder_path):
|
|||
|
|
for file_name in files:
|
|||
|
|
if file_name.endswith(".txt"):
|
|||
|
|
file_path = os.path.join(root, file_name)
|
|||
|
|
|
|||
|
|
# 确保文件可写
|
|||
|
|
make_writable(file_path)
|
|||
|
|
|
|||
|
|
# 读取文件内容并进行处理
|
|||
|
|
with open(file_path, "r") as file:
|
|||
|
|
lines = file.readlines()
|
|||
|
|
|
|||
|
|
processed_lines = []
|
|||
|
|
for line in lines:
|
|||
|
|
numbers = line.split()
|
|||
|
|
processed_numbers = []
|
|||
|
|
|
|||
|
|
# 确保第一列为整数 0 或 1，不处理为浮点数
|
|||
|
|
if (
|
|||
|
|
numbers[0] == "0"
|
|||
|
|
or numbers[0] == "1"
|
|||
|
|
or numbers[0] == "2"
|
|||
|
|
or numbers[0] == "3"
|
|||
|
|
or numbers[0] == "4"
|
|||
|
|
or numbers[0] == "5"
|
|||
|
|
or numbers[0] == "6"
|
|||
|
|
or numbers[0] == "7"
|
|||
|
|
or numbers[0] == "8"
|
|||
|
|
or numbers[0] == "9"
|
|||
|
|
or numbers[0] == "10"
|
|||
|
|
or numbers[0] == "11"
|
|||
|
|
or numbers[0] == "12"
|
|||
|
|
or numbers[0] == "13"
|
|||
|
|
or numbers[0] == "14"
|
|||
|
|
or numbers[0] == "15"
|
|||
|
|
or numbers[0] == "16"
|
|||
|
|
or numbers[0] == "17"
|
|||
|
|
or numbers[0] == "18"
|
|||
|
|
):
|
|||
|
|
processed_numbers.append(numbers[0])
|
|||
|
|
else:
|
|||
|
|
print(f"Unexpected value in first column: {numbers[0]}")
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
# 处理后面的列，保留原始格式并确保负数变成正数，且删除 NaN 数据
|
|||
|
|
skip_line = False # 用于标记是否跳过这一行
|
|||
|
|
for number in numbers[1:]:
|
|||
|
|
try:
|
|||
|
|
number = float(number)
|
|||
|
|
if math.isnan(number): # 检查是否为NaN
|
|||
|
|
skip_line = True
|
|||
|
|
print(
|
|||
|
|
f"NaN detected in file: {file_path}, line: {line}"
|
|||
|
|
)
|
|||
|
|
break
|
|||
|
|
if number < 0:
|
|||
|
|
number = abs(number) # 将负数转换为正数
|
|||
|
|
processed_numbers.append(str(number)) # 保留原始格式
|
|||
|
|
except ValueError:
|
|||
|
|
processed_numbers.append(number) # 非数字列保持原样
|
|||
|
|
|
|||
|
|
# 如果该行没有NaN数据，则加入结果列表
|
|||
|
|
if not skip_line:
|
|||
|
|
processed_line = " ".join(processed_numbers)
|
|||
|
|
processed_lines.append(processed_line)
|
|||
|
|
|
|||
|
|
# 将处理后的内容写回文件
|
|||
|
|
with open(file_path, "w") as file:
|
|||
|
|
file.write("\n".join(processed_lines))
|
|||
|
|
print(f"Finished processing: {file_path}")
|
|||
|
|
|
|||
|
|
# 指定文件夹路径
|
|||
|
|
folder_path = r"G:\dataset\PCS\before\labels"
|
|||
|
|
#run the function
|
|||
|
|
process_files_in_folder(folder_path)
|