首页 > 解决方案 > 如何使用 lambda 和 hasattr 过滤嵌套属性上的对象数组?

问题描述

我有以下 Python 代码:

myArray = [{ "id": 1, "desc": "foo", "specs": { "width": 1, "height": 1}}, { "id": 2, "desc": "bar", "specs": { "width": 2, "height": 2, "color": "black"}}, { "id": 3, "desc": "foobar"}]
print len(myArray)

myArray_filtered = filter(lambda item : hasattr(item, "specs") and hasattr(item.specs, "color"), myArray)
print len(myArray_filtered)

我希望1在第二次打印时获得长度,但它是0. 你能告诉我我的代码有什么问题吗?

标签: pythonarraysobjectlambdahasattr

解决方案


鉴于您的嵌套结构,您可以使用dict.get一些默认值:

>>> myArray_filtered = list(filter(lambda d: d.get("specs", {}).get("color") is not None, myArray))
>>> len(myArray_filtered)
1
>>> myArray_filtered
[{'id': 2, 'desc': 'bar', 'specs': {'width': 2, 'height': 2, 'color': 'black'}}]

推荐阅读