Ver Fonte

fix: correct row indexing for CSV translation starting from row 2

mrh (aider) há 1 ano atrás
pai
commit
5ba3224cc8
1 ficheiros alterados com 4 adições e 4 exclusões
  1. 4 4
      excel2sql/translate_csv.py

+ 4 - 4
excel2sql/translate_csv.py

@@ -12,8 +12,8 @@ def translate_csv(input_file):
     # 在B列右边插入一列空列
     df.insert(2, 'C', '')
 
-    # 获取B列从第三行开始的数据
-    data_to_translate = df.iloc[2:, 1].tolist()
+    # 获取B列从第二行开始的数据(跳过标题行)
+    data_to_translate = df.iloc[1:, 1].tolist()
 
     # 初始化翻译器
     translator = OpenAITranslator()
@@ -21,8 +21,8 @@ def translate_csv(input_file):
     # 批量翻译
     translated_texts = translator._batch_translate(data_to_translate)
 
-    # 将翻译结果写入C列,从第行开始
-    df.iloc[2:, 2] = translated_texts
+    # 将翻译结果写入C列,从第行开始
+    df.iloc[1:, 2] = translated_texts
 
     # 保存修改后的CSV文件
     output_file = input_file.replace('.csv', '_translated.csv')