【Python】输出最小k行的序号

# -*- coding:utf-8 -*-
# @FileName :0930_1337.py
# @DateTime :2021/9/30 13:10
# @Author   :wanglb
# 输出最小k行的序号
# lis = [
#     [1, 1, 0, 0, 0],
#     [1, 1, 1, 1, 0],
#     [1, 0, 0, 0, 0],
#     [1, 1, 0, 0, 0],
#     [1, 1, 1, 1, 1]
# ]
# k = 3

lis = [
    [1, 0, 0, 0],
    [1, 1, 1, 1],
    [1, 0, 0, 0],
    [1, 1, 0, 0]
]
k = 2
result = [each[0] for each in sorted(enumerate(map(sum, lis)), key=lambda x: x[1])[:k]]
print(result)

输出结果为 [0, 2]