Append 到 for 循环和 if 语句中的空列表

Question

The program should read the data in the PDB file named 1A36.pdb, and for each distinct atom in it, store a record for that atom in an list of atom records.

I am trying to append 'line' into empty list 'atom_records' within the for loop. When I print 'atom_records' I get duplicated records in scattered pattern, not sure why that is happening.

Expected Output when printing 'atom_records':

https://i.stack.imgur.com/sfQsU.png

My code: Ignore the commented parts, that is part of the for loop to print count of atom records.

import sys


def atom_info(count):
    file = open(sys.argv[1], "r")
    contents = file.read()
    lines = contents.splitlines()
    atom_records = []
    for line in lines:
        if line.startswith('ATOM'):
            count += 1
            atom_records.append(line)
            print(atom_records)
    return count

#count = 0
#frequency = atom_info(count)
#print("Welcome to the pdb program. \nTo begin, try typing 'help' for #the list of valid commands.")
#print(frequency, "atoms recorded.")

Answer 1

here my answer:

import sys


def atom_info(count):
    # file = open(sys.argv[1], "r")
    file = open('1a36.pdb' ,"r")
    contents = file.read()
    lines = contents.splitlines()
    atom_records = []
    for line in lines:
        if line.startswith('ATOM'):
            count += 1
            atom_records.append(line)
            # print(atom_records)     #delete here as suggested
            print(line)
            
    for i in atom_records:
        print(i)    #prints like : ATOM   5230  OXT PHE A 765      35.802  38.654  28.765  1.00 36.28           O  
        
    # print(atom_records) # prints like : 'ATOM   5229  CZ  PHE A 765      32.510  34.679  27.197  1.00 30.57           C  ', 
    return count

print('N# atoms : ',atom_info(0))