if文を使用した特定行の読み込み

Question

A 1.2 3.6 2.1
B 2.2 -0.4 6.1
C 2.9 -4.2 0
D 3.4 4.9 9.4
E 4.8 1.2 3.2
A 1.1 3.8 9.1
B -2.1 3.4 5.1
C 1.2 0.9 5.4
D 2.5 0.3 3.4
E 2.3 2.4 2.5
・
・
・

このようにABCのxyz座標が続いている文字があったとして、AB、AD、AE間の距離を計算したい場合
if文を使って、Cの座標のみ読み込まないようなプログラムを書きたいのですが、どのように書けば
よいでしょうか？
このようなプログラムを書きましたがうまくいきませんでした。

name = [""]*line_number
xxx_coord = [0.0]*line_number
yyy_coord = [0.0]*line_number
zzz_coord = [0.0]*line_number


for k in range(atom_number):
line  = input_file.readline()
items = line.split()
atom_name[k] = items[0]
xxx_coord[k] = float( items[1] )
yyy_coord[k] = float( items[2] )
zzz_coord[k] = float( items[3] )
    k =+ 1

for i in range(line_number): 
   if name == "C":
           continue
   for j in range(line_number):
        if atom_name == "C":
              continue
        dx = xxx_coord[i] - xxx_coord[j]
        dy = yyy_coord[i] - yyy_coord[j]
        dz = zzz_coord[i] - zzz_coord[j]
        
        square_r = dx*dx + dy*dy + dz*dz
        
        r = math.sqrt( square_r )

あくまで、if文を使用して計算を行いたいです。
よろしくお願いします。

Comments

[manzyun]

恐れ入ります。この質問文のプログラムのコードのはじめと終わりに 「```（半角ダッシュ3つ）」を追記いただき、インデントをつけて、もともとのソースコードがどのようになっているのか分かりやすく変更お願いいたします。

[crmy]

変更しました、よろしくお願いいたします。

Answer 1

もとの質問はインデントが崩れてしまっているので、解読できません。
こんな感じのアウトプットを期待されているのでしょうか？

import math

class Coordinate():
    def __init__(self, x, y, z):
        self.x = x
        self.y = y
        self.z = z

    def compute_distance(self, other):
        dx = self.x - other.x
        dy = self.y - other.y
        dz = self.z - other.z
        return math.sqrt(dx**2 + dy**2 + dz**2)

    def __str__(self):
        return 'x: {0}, y: {1}, z:{2}'.format(self.x, self.y, self.z)

def compute_distance(coord1, coord2):
    return coord1.compute_distance(coord2)

def main():
    with open('data.txt') as f:
        coord_dict = {}

        for line in f:
            elems = line.split()
       
            key = elems[0]
            values = list(map(float, elems[1:]))

            coord_dict[key] = Coordinate(*values)

            if key == 'E':
                print('-' * 50)
                print('A to B', compute_distance(coord_dict['A'], coord_dict['B']))
                print('A to D', compute_distance(coord_dict['A'], coord_dict['D']))
                print('A to E', compute_distance(coord_dict['A'], coord_dict['E']))

if __name__ == '__main__':
    main()

"""出力
--------------------------------------------------
A to B 5.744562646538029
A to D 7.734339015067804
A to E 4.464302857109943
--------------------------------------------------
A to B 5.138093031466052
A to D 6.833739825307955
A to E 6.8527366796047255
"""

NumPyを使ったらもっと楽になりそうですが。

import numpy as np

def main():
    with open('data.txt') as f:
        coord_dict = {}

        for line in f:
            elems = line.split()
       
            key = elems[0]
            values = np.array(elems[1:], dtype=np.float)

            coord_dict[key] = values

            if key == 'E':
                print('-' * 50)
                print('A to B', np.linalg.norm(coord_dict['A'] - coord_dict['B']))
                print('A to D', np.linalg.norm(coord_dict['A'] - coord_dict['D']))
                print('A to E', np.linalg.norm(coord_dict['A'] - coord_dict['E']))

if __name__ == '__main__':
    main()

"""出力
--------------------------------------------------
A to B 5.74456264654
A to D 7.73433901507
A to E 4.46430285711
--------------------------------------------------
A to B 5.13809303147
A to D 6.83373982531
A to E 6.8527366796
"""

もっといい方法がありそうな気もしますが、とりあえずこんな感じで。
pandasとかpandasとかpandasとか。