De retour de sous-ensemble de la Liste qui Correspond à Condition

Disons que j'ai une liste de ints:

listOfNumbers = range(100)

Et je veux retourner une liste des éléments qui répondent à certaines conditions, de dire:

def meetsCondition(element):
    return bool(element != 0 and element % 7 == 0)

Qu'est ce qu'un Pythonic retour dans un sous-list de l'élément dans un list pour qui meetsCondition(element) est True?

Une approche naïve:

def subList(inputList):
    outputList = []

    for element in inputList:
        if meetsCondition(element):
            outputList.append(element)

    return outputList

divisibleBySeven = subList(listOfNumbers)

Est-il une façon simple de le faire, peut-être avec une compréhension de liste ou set() fonctions, et sans le temporaire outputList?

Notez que divisibleBySeven est en fait la même list objet comme outputList, ce n'est pas une copie de celui-ci.

OriginalL'auteur Will | 2016-01-30

  1. 24

    Utiliser compréhension de liste,

    divisibleBySeven = [num for num in inputList if num != 0 and num % 7 == 0]

    ou vous pouvez utiliser le meetsCondition aussi,

    divisibleBySeven = [num for num in inputList if meetsCondition(num)]

    vous pouvez réellement écrire les mêmes conditions qu'avec Python truthy sémantique, comme ce

    divisibleBySeven = [num for num in inputList if num and num % 7]

    alternativement, vous pouvez utiliser filter fonction avec votre meetsCondition, à l'instar de ce

    divisibleBySeven = filter(meetsCondition, inputList)
    En Python 3, filter retourne un itérateur (OIE un générateur), donc si vous avez réellement besoin d'une liste, alors vous devez faire quelque chose comme list(filter(meetsCondition, inputList)).
    Vrai que. Je ne comprend pas que, parce que l'OP tagged 2.7 😀
    Juste assez. Je viens de comprendre, il peut être utile d'info pour les futurs lecteurs.

    OriginalL'auteur thefourtheye