如何编写能够读取.xyz文件并计算原子之间距离的代码？

def calculate_distance(atom1, atom2):
    # 计算两个原子之间的距离
    distance = ((atom1[0] - atom2[0])**2 + (atom1[1] - atom2[1])**2 + (atom1[2] - atom2[2])**2)**0.5
    return distance

def read_xyz_file(file_path):
    atoms = []
    with open(file_path, 'r') as file:
        lines = file.readlines()
        num_atoms = int(lines[0])
        for line in lines[2:num_atoms+2]:
            atom_info = line.split()
            atom_name = atom_info[0]
            atom_coords = [float(coord) for coord in atom_info[1:4]]
            atoms.append((atom_name, atom_coords))
    return atoms

def main():
    file_path = 'example.xyz'  # 替换为实际的.xyz文件路径
    atoms = read_xyz_file(file_path)
    num_atoms = len(atoms)
    
    for i in range(num_atoms):
        for j in range(i+1, num_atoms):
            atom1 = atoms[i]
            atom2 = atoms[j]
            distance = calculate_distance(atom1[1], atom2[1])
            print(f"Distance between {atom1[0]} and {atom2[0]}: {distance:.2f} units")

if __name__ == '__main__':
    main()