数据样本表
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()
我有这个功能,它根据过程,班次和级别返回单位和小时,具体取决于数据类型是单位还是数据类型是小时。我将如何创建第二个函数,该函数根据也使用第一个函数的数据类型,根据过程,班次和级别来计算单位/小时的速率。