model_names = ['is1', 'is5', 'is10', 'im1', 'im5', 'im10']
举个例子
models = [0.1, 0.2, 0.1, 0.3, 0.2, 0.3]
列表的最大值为0.3,对应于型号名称“im1”和“im10”。我想要这个功能
dominant_model(models)
返回
['im1', 'im10']
即,最“主导”的模式。你知道吗
这是我的代码,但我想知道是否有一个更简短的方法来做同样的事情,而不是所有的条件:
def dominant_model(models):
m = max(models)
dom_models = [i for i,j in enumerate(models) if j==m]
for i in range(len(dom_models)):
if dom_models[i]==0:
dom_models[i]=model_names[0]
elif dom_models[i]==1:
dom_models[i]=model_names[1]
elif dom_models[i]==2:
dom_models[i]=model_names[2]
elif dom_models[i]==3:
dom_models[i]=model_names[3]
elif dom_models[i]==4:
dom_models[i]=model_names[4]
elif dom_models[i]==5:
dom_models[i]=model_names[5]
return dom_models
在找到max
m = max(model)
之后,将模型名称压缩到值中,并获取值等于m
的名称。你知道吗演示
使用两个循环:
相关问题 更多 >
编程相关推荐