Reputation: 469
I have the following data in my list of dictionary:
data = [{'I-versicolor': 0, 'Sepal_Length': '7.9', 'I-setosa': 0, 'I-virginica': 1},
{'I-versicolor': 0, 'I-setosa': 1, 'I-virginica': 0, 'Sepal_Width': '4.2'},
{'I-versicolor': 2, 'Petal_Length': '3.5', 'I-setosa': 0, 'I-virginica': 0},
{'I-versicolor': 1.2, 'Petal_Width': '1.2', 'I-setosa': 0, 'I-virginica': 0}]
And to get a list based upon a key and value I am using the following:
next((item for item in data if item["Sepal_Length"] == "7.9"))
However, all the dictionary doesn't contain the key Sepal_Length
, I am getting :
KeyError: 'Sepal_Length'
How can i solve this?
Upvotes: 3
Views: 4754
Reputation: 309821
You can use dict.get
to get the value:
next((item for item in data if item.get("Sepal_Length") == "7.9"))
dict.get
is like dict.__getitem__
except that it returns None
(or some other default value if provided) if the key is not present.
Just as a bonus, you don't actually need the extra parenthesis here around the generator expression:
# Look mom, no extra parenthesis! :-)
next(item for item in data if item.get("Sepal_Length") == "7.9")
but they help if you want to specify a default:
next((item for item in data if item.get("Sepal_Length") == "7.9"), default)
Upvotes: 7