def distance (x1,y1,x2,y2): #функция для вычисления расстояния между координатами
   dist = ((((x2-x1)**2) + ((y2-y1)**2))**(1/2))
   return dist
def spisok (stroka):
    stroka = stroka.replace(',', '')
    stroka = stroka.replace('[', '')
    stroka = stroka.replace(']', '')
    stroka = stroka.split()
    stroka = list(map(float, stroka))
    return stroka
# import pandas as pd
# import numpy as np
import csv
i=0
pogr = []


with open('2.csv', 'r') as csvFile: #открываю файл из архива с данными
    dictReader = csv.DictReader(csvFile)
    for key in dictReader:
        if i<=319:  # 1.1 1.2
            a= spisok(((key['GPS'])))
            el = distance(55.642242,37.39408,a[0],a[1])
            pogr.append(el)
            i+=1
        elif (i > 319) and (i <= 668): #2.1 2.2
            a = spisok(((key['GPS'])))
            el = distance(55.604025,37.41843,a[0],a[1])
            pogr.append(el)
            i+=1
        elif (i<=824)and (i > 668):#2.1 2.2
            a = spisok(((key['GPS'])))
            el = distance(55.547298, 37.482864,a[0],a[1])
            pogr.append(el)
            i+=1
# pogr.insert(0,'pogreshnost')

with open('2.csv', 'r') as csvFile:
    infileReader = csv.reader(csvFile)
    rows = list(infileReader)

header = rows[0]
header.append("Distance")

for i in range(1, len(rows)):
    rows[i].append(pogr[i])

with open('2.csv', 'w', newline='') as outputFile:
    writer = csv.writer(outputFile)
    writer.writerows(rows)
# print(pogr)

# from itertools import zip_longest
# r=zip(pogr)
# file=open('2.csv', 'a')
# w = csv.writer(file)
# for row in r:
#         w.writerow(row)
# file.close()
# # ser = pd.Series(pogr,index=file.index)
# # file.loc[:, 'pogr'] = ser
# # df['pogr'] = pogr