Python获取嵌套列表首元素最大值对应索引报错排查

你的代码存在几个关键错误，导致报错且功能不符合预期：

1. 核心逻辑错误 #

你写的[similarity_list[1]]只取了列表中索引为1的单个子列表，求max后得到的是[0, 0]，完全没有遍历所有子列表的首元素，这和你“获取所有子列表首元素最大值对应索引”的需求完全不符。

2. 报错原因 #

出现The truth value of an array with more than one element is ambiguous错误，大概率是因为你的similarity_list不是普通Python嵌套列表，而是numpy数组。numpy数组在进行元素匹配（比如index方法里的查找）时，会尝试对整个数组做布尔判断，数组元素多于一个时就会触发这个歧义错误，这时候用any()或all()根本不对症，因为问题出在逻辑方向上。

3. 正确实现方法 #

情况一：如果是普通Python嵌套列表 #

# 提取所有子列表的首元素
first_elements = [item[0] for item in similarity_list]
# 找到首元素的最大值
max_val = max(first_elements)
# 获取所有最大值对应的索引
indices = [i for i, val in enumerate(first_elements) if val == max_val]

运行后会得到[6, 11]，正好对应你列表中两个首元素为1.0的子列表索引。

情况二：如果是numpy数组 #

import numpy as np

# 提取首元素列
first_elements = similarity_list[:, 0]
# 找到最大值
max_val = first_elements.max()
# 获取所有对应索引
indices = np.where(first_elements == max_val)[0]

同样会得到array([ 6, 11])，符合需求。

另外补充：就算用普通列表，list.index()方法也只能返回第一个匹配元素的索引，没法获取所有符合条件的索引，所以必须用列表推导式或numpy的where来实现“所有索引”的需求。

内容的提问来源于stack exchange，提问作者Dario Federici