Python 3 - Вычислить среднее значение и запись в .csv

Question

(А) Python код

import csv 
from collections import defaultdict 

data = defaultdict(str) 

#Make a list with the predefined variables 
definition = ["record_id", "abbreviation", "patient_id", "study_id", 
"step_count", "distance", "ambulation_time", "velocity", "cadence", 
"normalized_velocity", "step_time_differential", "step_length_differential", 
"cycle_time_differential", "step_time", "step_length", "step_extremity", 
"cycle_time", "stride_length", "hh_base_support", "swing_time", 
"stance_time", "single_supp_time", "double_supp_time", "toe_in_out"] 

#Read the GaitRite .csv 
with open('C:/Users/Kay_v/Documents/School/Exports/Export 3.csv', 'r') as f, open('C:/Users/Kay_v/Documents/School/Exports/result.csv', 'w') as outfile: 
    reader = csv.reader(f, delimiter=';') 
    next(reader, None) # skip the headers 
    writer = csv.DictWriter(outfile, fieldnames=definition, lineterminator='\n') 
    writer.writeheader() 

#Read the .csv row by row 
    for row in reader: 
     #print(row) 
     for item in definition: 
      h = item.replace('_', '') 
      r0 = row[0].lower().replace(' ', '') 
      if h in r0: 
       try: 
        avg = round((float(row[1].replace(',', '.')) + float(row[2].replace(',', '.')))/2, 2) 
      except ValueError: 
       avg = 0 # for cases with entry strings or commas 
       #print(avg) 
       print(h, r0, row[1], row[2]) 
       data[item] = row[1] 

    data['record_id'] = 1 

# Write the clean result.csv 
    writer.writerow(data) 

(Б) Проблема

Проблема состоит в том числе о средних значений в Result.csv , Я использую следующую часть кода для вычисления среднего значения, когда переменная имеет два значения. В нынешней ситуации средняя рассчитывается, но он не отображается в Result.csv

try: 
    avg = round((float(row[1].replace(',', '.')) + float(row[2].replace(',', '.')))/2, 2) 
except ValueError: 
    avg = 0 # for cases with entry strings or commas 

Я надеюсь, что кто-то может помочь, чтобы получить среднее значение, чтобы показать в Result.csv Aswell, был бы весьма признателен!

Не стесняйтесь играть с файлом экспорта я использую, вы можете скачать его здесь: CSV export file

Answer 1

Попробуйте это:

if h in r0: 
    try: 
     avg = round((float(row[1].replace(',', '.').replace(';', '.')) + float(row[2].replace(',', '.').replace(';', '.')))/2, 2) 
     data[item] = avg 
    except ValueError: 
     data[item] = 0 # for cases with entry strings or commas 
     #print(avg) 
     print(h, r0, row[1], row[2]) 

Answer 2

Вы вычисления среднего, но не добавляя его в файл CSV файл первой проблемы, с которой вы сталкиваетесь, из того, что я понял из вашего двусмысленного вопроса. Сначала добавьте еще один столбец в списке definition и добавить ключ с именем average в Словаре data, вот модифицированный код:

import csv 
from collections import defaultdict 

data = defaultdict(str) 

#Make a list with the predefined variables 
definition = ["record_id", "abbreviation", "patient_id", "study_id", 
"step_count", "distance", "ambulation_time", "velocity", "cadence", 
"normalized_velocity", "step_time_differential", "step_length_differential", 
"cycle_time_differential", "step_time", "step_length", "step_extremity", 
"cycle_time", "stride_length", "hh_base_support", "swing_time", 
"stance_time", "single_supp_time", "double_supp_time", "toe_in_out", "average"] 

#Read the GaitRite .csv 
with open('Export 3.csv', 'r') as f, open('result.csv', 'w') as outfile: 
    reader = csv.reader(f, delimiter=';') 
    next(reader, None) # skip the headers 
    writer = csv.DictWriter(outfile, fieldnames=definition, lineterminator='\n') 
    writer.writeheader() 

#Read the .csv row by row 
    for row in reader: 
     #print(row) 
     for item in definition: 
      h = item.replace('_', '') 
      r0 = row[0].lower().replace(' ', '') 
      if h in r0: 
       try: 
        avg = round((float(row[1].replace(',', '.')) + float(row[2].replace(',', '.')))/2, 2) 
       except ValueError: 
        avg = 0 # for cases with entry strings or commas 
       # print(avg) 
       # print(h, r0, row[1], row[2]) 
       data[item] = row[1] 
       data['average'] = avg  
    data['record_id'] = 1 

# Write the clean result.csv 
    print data 
    writer.writerow(data) 

Answer 3

Я попытаюсь объяснить это лучше. Я хотел бы, чтобы в конечном итоге Result.csv выход следующее:

Желаемая выход

record_id abbreviation patient_id study_id step_count distance ambulation_time velocity cadence normalized_velocity step_time_differential step_length_differential cycle_time_differential step_time step_length step_extremity cycle_time stride_length hh_base_support swing_time stance_time single_supp_time double_supp_time toe_in_out 
1            3   292,34 1,67    175,1  107,8       0,004     1,051      0,008     0,56  97,27      1,11  194,64   4,65    0,47  0,65   0,47    0,18    1,45 

Проблема заключается в том, что некоторые из значений в CSV-я чтение данных из, содержит 2 значения (например, step_time [0,558; 0,554]), а другие содержат только 1 значение (например, step_count [3]). Те, которые содержат только 1 значение, могут быть сразу переданы в result.csv. Но для тех, которые содержат 2 значения, среднее значение этих двух значений должно быть рассчитано, а затем это значение также должно быть передано в result.csv