Извлечение данных из списка с использованием цикла for

Question

Я довольно новичок в python. Для упражнения я читаю данные из файла csv и сохраняю его в списке. Возникли проблемы с доступом к элементам в списке и последующим их сохранением.

Данные, содержащиеся в файле будет идентификатор, а затем двумя наборами координат

ID, x1, y1, x2, y2 

Мне нужно, чтобы иметь возможность извлекать данные из моего списка, которые будут использоваться в расчетах позже. надеясь перевернуть список для нескольких строк, которые будут вычислены.

Любая помощь приветствуется!

Приветствия

Answer 1

Вы должны использовать csv библиотеку

with open('/path/to/file.csv', 'r') as f: 
    reader = csv.reader(f) 
    for row in reader: 
     ID, x1, y1, x2, y2 = row 

Answer 2

Одним из подходов является сбор данных в словаре (возможно id поле может быть ключом) и собрать все координаты для конкретного идентификатора ,

Вот один такой подход:

import csv 
from collections import defaultdict 

data = defaultdict(list) 

with open('your_file.csv') as f: 
    reader = csv.reader(f) 
    next(reader) # skips the header row 
    for row in reader: 
    data[row[0]].append(((row[1],row[2]),(row[3],row[4]))) 

---

Основываясь на ваш комментарий:

Как бы я тогда получить доступ к значениям, чтобы вычислить следующее: SQRT ((x2-x1) ** 2 + (y2-y1) ** 2) для каждой из двух строк (пример 1 и пример2), содержащиеся в словаре?

Вы можете сделать расчеты, как вы читаете файл, а просто хранить результаты по id значение:

for row in reader: 
    id, x1, y1, x2, y2 = row 
    # We have to convert the numbers from a string to floats for the calculation 
    result = sqrt(float(x2)-float(x1))**2 + (float(y2)-float(y1))**2 
    data[id].append(result) 

Затем, вы можете просто распечатать результаты:

for id, results in data.iteritems(): 
    print('For ID: {}'.format(id)) 
    for row, result in enumerate(results): 
     print('\t{} - {}'.format(row+1, result))