58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
|
import csv
|
|||
|
|
|
|||
|
|
def format_value(val):
|
|||
|
|
"""格式化值：整数保持原样，小数保留3位"""
|
|||
|
|
try:
|
|||
|
|
num_val = float(val)
|
|||
|
|
if num_val.is_integer():
|
|||
|
|
return int(num_val)
|
|||
|
|
return round(num_val, 3)
|
|||
|
|
except ValueError:
|
|||
|
|
return val
|
|||
|
|
|
|||
|
|
def csv_to_ini(csv_file, ini_file):
|
|||
|
|
"""
|
|||
|
|
将CSV文件转换为INI格式
|
|||
|
|
:param csv_file: 输入的CSV文件路径
|
|||
|
|
:param ini_file: 输出的INI文件路径
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
with open(csv_file, 'r', newline='', encoding='utf-8') as csv_input:
|
|||
|
|
# 读取CSV文件
|
|||
|
|
reader = csv.DictReader(csv_input)
|
|||
|
|
|
|||
|
|
# 确定所需字段
|
|||
|
|
required_fields = ['name', 'x', 'y', 'z', 'u', 'v', 'w',
|
|||
|
|
'id', 'order', 'lineid', 'status', 'linetype']
|
|||
|
|
|
|||
|
|
with open(ini_file, 'w', encoding='utf-8') as ini_output:
|
|||
|
|
# 处理每一行数据
|
|||
|
|
for row in reader:
|
|||
|
|
# 检查是否所有必需字段都存在
|
|||
|
|
if not all(field in row for field in required_fields):
|
|||
|
|
missing = [field for field in required_fields if field not in row]
|
|||
|
|
print(f"警告: 行 {reader.line_num} 缺少字段 {', '.join(missing)}，跳过该行")
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
# 写入section头
|
|||
|
|
ini_output.write(f"[{row['name']}]\n")
|
|||
|
|
|
|||
|
|
# 写入数值字段
|
|||
|
|
for field in required_fields[1:]: # 跳过name字段
|
|||
|
|
formatted = format_value(row[field])
|
|||
|
|
ini_output.write(f"{field} = {formatted}\n")
|
|||
|
|
|
|||
|
|
# 区块间添加空行
|
|||
|
|
ini_output.write("\n")
|
|||
|
|
|
|||
|
|
print(f"转换成功！INI文件已保存至: {ini_file}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"处理过程中出错: {str(e)}")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
# 在此输入您的文件路径
|
|||
|
|
input_csv = "D:/aa.csv" # 替换为实际CSV文件路径
|
|||
|
|
output_ini = "D:/output.ini" # 输出的INI文件路径
|
|||
|
|
|
|||
|
|
csv_to_ini(input_csv, output_ini)
|