如何对txt文件中的cloumn进行重新排序？

# 读取txt文件
with open('data.txt', 'r') as file:
    content = file.readlines()

# 解析txt文件内容为列表
data = [line.strip().split('\t') for line in content]

# 获取column索引
header = data[0]
column_index = {column: index for index, column in enumerate(header)}

# 按照指定顺序重新排序column
new_order = ['column3', 'column1', 'column2']
sorted_header = sorted(header, key=lambda x: new_order.index(x))

# 更新数据结构中的column顺序
for row in data:
    row[:] = [row[column_index[column]] for column in sorted_header]

# 将更新后的内容写回txt文件
with open('data.txt', 'w') as file:
    for row in data:
        file.write('\t'.join(row) + '\n')