在python中从制表符分隔的文本文件中选择列，并将其添加到tsv文件中

import csv
import os

input_file = "input.txt"
output_file = "output.tsv"

with open(input_file, "r", newline="") as file:
    reader = csv.reader(file, delimiter="\t")

selected_columns = [0, 2]  # 选择第1列和第3列

with open(output_file, "w", newline="") as file:
    writer = csv.writer(file, delimiter="\t")

for row in reader:
    selected_data = [row[i] for i in selected_columns]
    writer.writerow(selected_data)

print("列已成功添加到tsv文件中。")

import csv
import os

input_file = "input.txt"
output_file = "output.tsv"

with open(input_file, "r", newline="") as file:
    reader = csv.reader(file, delimiter="\t")

    selected_columns = [0, 2]  # 选择第1列和第3列

    with open(output_file, "w", newline="") as file:
        writer = csv.writer(file, delimiter="\t")

        for row in reader:
            selected_data = [row[i] for i in selected_columns]
            writer.writerow(selected_data)

print("列已成功添加到tsv文件中。")