数据样本表

import csv
f = open('C:\\Users\\keshabg\\Desktop\\sql_testing\\table_1.csv')
csv_f = csv.reader(f)

def try1(data,process,shift,lc):
    result = []
    if data == 'units':
        result_column = 5
    elif data == 'hours':
        result_column = 6
    else:
        raise NotImplementedError
    for row in csv_f:
        if row[2] == shift and row[3] == process and row[4] == lc:
            result.append(row[result_column])
    return result

if __name__=="__try1__":
    try1()

我有这个功能，它根据过程，班次和级别返回单位和小时，具体取决于数据类型是单位还是数据类型是小时。我将如何创建第二个函数，该函数根据也使用第一个函数的数据类型，根据过程，班次和级别来计算单位/小时的速率。