我正在使用Python的max和min函数对列表进行minimax算法，我需要max()或min()返回值的索引。换句话说，我需要知道哪一步棋产生了最大(第一个玩家的回合)或最小(第二个玩家的回合)值。
for i in range(9):
new_board = current_board.new_board_with_move([i / 3, i % 3], player)
if new_board:
temp = min_max(new_board, depth + 1, not is_min_level)
values.append(temp)
if is_min_level:
return min(values)
else:
return max(values)
我需要能够返回最小值或最大值的实际索引，而不仅仅是值。
当前回答
我认为最好的方法是将列表转换为numpy数组并使用这个函数:
a = np.array(list)
idx = np.argmax(a)
其他回答
if is_min_level:
return values.index(min(values))
else:
return values.index(max(values))
只要你知道如何使用lambda和"key"参数，一个简单的解决方案是:
max_index = max( range( len(my_list) ), key = lambda index : my_list[ index ] )
如果你不想导入额外的模块，找到列表中值最小的索引的简单方法:
min_value = min(values)
indexes_with_min_value = [i for i in range(0,len(values)) if values[i] == min_value]
然后选择第一个例子:
choosen = indexes_with_min_value[0]
就这么简单:
stuff = [2, 4, 8, 15, 11]
index = stuff.index(max(stuff))
如果枚举列表中的项目，可以同时找到min/max索引和值，但对列表的原始值执行min/max。像这样:
import operator
min_index, min_value = min(enumerate(values), key=operator.itemgetter(1))
max_index, max_value = max(enumerate(values), key=operator.itemgetter(1))
这样，对于min(或max)，列表将只遍历一次。
