使用列表理解创建 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 = 12assert 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