Reorganizar os arquivos de dados com a inclusão de números de sequência

Question

Eu estou em apuros tentando escrever um script para reorganizar o meu arquivo e a esperança de que alguém aqui pode ajudar.Folheei o estouro de pilha e encontrado algum bom programa para resolver o meu problema.Infelizmente, eu ainda enfrentam alguns problemas que precisam de ajuda.

Inicialmente eu tenho um arquivo que contém:

A X1 X2 X3
B X5 X6 X7

Eu quero que o arquivo a ser como este:

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

Eu tento código-lo assim, mas eu não sei como incluir o 1, 2, 3 em cada linhas como afirmado acima:

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

Alguém pode me ensinar como fazer isso?

Answer 1

Há uma útil função interna chamada enumerate():

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