如何找到元素总数最小的列索引。我想找到使用lambda的简短解决方案。例如对于给定的矩阵：
[[7, 2, 7, 2, 8],
[2, 9, 4, 1, 7],
[3, 8, 6, 2, 4],
[2, 5, 2, 9, 1],
[6, 6, 5, 4, 5]]
它应该返回3。因为第三列的元素总和最小。
使用numpy，您可以sum按列，并使用argmin()来找出最小总和索引：
import numpy as np
np.array(lst).sum(axis = 0).argmin()
# 3
或者，您可以不使用numpy包而使用列表推导：
min(((i, sum(v)) for i, v in enumerate(zip(*lst))), key = lambda x: x[1])[0]
# 3
zip(*lst)转置列表本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句