Python 3 - 计算平均值并写入 .csv

Question

(A) 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)

(B) 问题

问题在于将平均值包含在 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 中，将不胜感激！

随意使用我正在使用的导出文件，您可以在这里下载： 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的键添加到dictdata中，这是修改后的代码：

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 个值的值，应计算这 2 个值的平均值，然后将该值也传递给 result.csv