使用列表理解创建 def

Question

我尝试使用列表推导而不是标准循环来在 python 中定义一个函数。在这里，我想编写一个函数 ，get_index(CS, alpha)它将列表 CS 和阈值 alpha 作为输入，并返回第一个元素的位置索引，例如CS[index] > alpha。

例如：

CS = [0, 0, 0, 0, 1, 0, 1, 4, 9, 8, 12, 13, 12, 10, 9, 7, 9]

alpha = 12

assert get_index(CS, alpha) == 11

如果在 CS 中没有找到这样的元素，该函数应该返回 -1。

我的以下代码由于SyntaxError: invalid syntax.

def get_index(CS, alpha):

    return i for i, v in enumerate(CS) if v > alpha

    return -1

但如果我把它放在一个标准循环中，它就可以工作，比如：

def get_index(CS, alpha):

    for i, v in enumerate(CS):

        if v > alpha: 

            return i

    return -1

我无法区分并试图找出原因。欣赏你的想法。谢谢你。

Answer 1

如果您真的希望为此使用列表推导，请执行以下操作：def get_index(CS, alpha):    l = [i for i, v in enumerate(CS) if v > alpha] or [-1]    return l[0]尽管出于多种原因，您的第二个代码要好得多。发生语法错误，因为不允许在函数内有两个无条件返回。

Answer 2

您可以使用生成器表达式，它类似于列表推导式，但不会创建列表。调用next评估生成器，直到找到第一个结果，StopIteration如果没有，则引发。try:    return next(i for i, v in enumerate(CS) if v > alpha)except StopIteration:    return -1