在Python中,可以使用csv
模块来处理制表符分隔的文本文件,并将选择的列添加到tsv文件中。
首先,需要导入csv
模块和os
模块(用于文件路径操作):
import csv
import os
接下来,可以定义输入文件和输出文件的路径:
input_file = "input.txt"
output_file = "output.tsv"
然后,可以使用csv.reader
来读取输入文件,并使用制表符作为分隔符:
with open(input_file, "r", newline="") as file:
reader = csv.reader(file, delimiter="\t")
接着,可以选择要添加到tsv文件中的列。假设要选择第1列和第3列,可以使用列表索引来获取这些列的数据:
selected_columns = [0, 2] # 选择第1列和第3列
然后,可以创建一个输出文件,并使用csv.writer
来写入数据到tsv文件中:
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文件中。")
这段代码会将输入文件中选择的列添加到输出文件中,并使用制表符作为分隔符。你可以根据实际需求修改selected_columns
列表来选择不同的列。
云+未来峰会
DBTalk
DB TALK 技术分享会
云+社区技术沙龙[第9期]
云+社区开发者大会 武汉站
DB TALK 技术分享会
云+社区技术沙龙[第8期]
Elastic 中国开发者大会
领取专属 10元无门槛券
手把手带您无忧上云