重新排列包含序列号的数据文件

时间:2014-02-24 22:11:09

标签: python file output

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

最初我有一个包含以下内容的文件:

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()

有人可以教我怎么做吗?

1 个答案:

答案 0 :(得分:3)

有一个方便的内置函数叫enumerate()

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