Is it possible to compare one variable to several other variables in a if statement in python?

Question

I would like to improve this following if statement

#Extract ORG, GPE, LOC and FAC labels from phrases
for entity in doc.ents:
    if entity.label_ == "ORG" or entity.label_ == "GPE" or entity.label_ == "LOC" or entity.label_ == "FAC":
        print(entity.text, entity.label_)

Is it possible reduce the number of "entity.label_" variables to one?

`if entity.label_ in ('ORG', 'GPE', 'LOC', 'FAC'):` – khelwood Jul 07 '20 at 08:48 — khelwood, Jul 07 '20 at 08:48

score 1 · Accepted Answer · answered Jul 07 '20 at 08:50

1

You can try to check if entity.label_ var in a tuple of all the words.

for entity in doc.ents:
    if entity.label_ in ("ORG", "GPE", "LOC", "FAC"):
        print(entity.text, entity.label_)

answered Jul 07 '20 at 08:50

Leo Arad

4,452
2
6
17

score 0 · Answer 2 · answered Jul 07 '20 at 08:52

0

for entity in doc.ents:
    if any(entity.label_ == x for x in ["ORG", "GPE", "LOC", "FAC"]):
        print(entity.text, entity.label_)

answered Jul 07 '20 at 08:52

Jasper Nichol M Fabella

460
2
12

Is it possible to compare one variable to several other variables in a if statement in python?

2 Answers2