Python | 获取大于 K 的第一个元素的索引

在处理数据时，我们经常需要对数据进行筛选，获取符合要求的数据。本文将介绍如何使用 Python 编写一个函数，获取列表中第一个大于 K 的元素的索引。

算法思路

我们可以遍历列表，使用一个变量 idx 记录当前元素的索引，如果当前元素大于 K，则返回当前索引。

代码实现

下面是获取大于 K 的第一个元素索引的示例代码：

def find_first_gt(lst, k):
    for idx, val in enumerate(lst):
        if val > k:
            return idx
    return -1

函数 find_first_gt 接受两个参数：lst 表示列表，k 表示目标元素。函数会遍历列表，查找第一个大于目标元素的索引。如果找到，返回该索引；如果未找到，返回 -1。

测试示例

我们可以编写一些测试用例来验证函数的正确性，下面是几个示例：

assert find_first_gt([1, 2, 3], 2) == 2
assert find_first_gt([1, 3, 5], 2) == 1
assert find_first_gt([1, 3, 5], 6) == -1

以上代码会依次测试函数在不同场景下的表现。如果函数没有问题，以上测试都将通过。

总结

本文介绍了如何使用 Python 编写一个函数，获取列表中第一个大于 K 的元素的索引。我们通过遍历列表，查找符合要求的元素，实现了这个功能。