python - 重新排列数据文件并包含序列号

Question

我在尝试编写脚本来重新排列我的文件时遇到麻烦，希望这里有人可以提供帮助。我浏览了堆栈溢出并找到了一些不错的脚本来解决我的问题。不幸的是，我仍然面临一些需要帮助的问题。

最初我有一个文件包含:

A X1 X2 X3
B X5 X6 X7

我希望文件是这样的:

A, 1, X1
A, 2, X2
A, 3, X3
B, 1, X5
B, 2, X6
B, 3, X7

我尝试这样编码，但我不知道如何在每行中包含 1、2、3，如上所述:

with open('filename.txt','r') as f:
file=open('filename_2.txt','w')
for line in f:
    line=line.rstrip().split(' ')
    for item in line[1:]:
        p=line[0], item,'\n'
        file.writelines(p)
file.close()

有人可以教我怎么做吗？

Answer 1

有一个方便的内置函数，名为 enumerate() :

for line in f:
    items = line.rstrip().split()
    for index, item in enumerate(items[1:]):
        file.write("{}, {}, {}\n".format(items[0], index, item)