У меня есть список словарей:
friends = [
{'name': 'Sam', 'gender': 'male', 'sport': 'Basketball'},
{'name': 'Emily', 'gender': 'female', 'sport': 'volleyball'},
]
Мне нужно создать функции query
, select
и field_filter
для работы с похожими списками. Эти функции должны предоставлять возможность выбирать необходимые столбцы и производить фильтрацию по этим столбцам.
Пример:
result = query(friends,
select('name', 'gender', 'sport'),
field_filter('sport', *('Basketball', 'volleyball')),
field_filter('gender', *('male',)))`
[{'gender': 'male', 'name': 'Sam', 'sport': 'Basketball'}]
Мне нужно использовать предопределенный код:
from typing import Dict, Any, Callable, Iterable
DataType = Iterable[Dict[str, Any]]
ModifierFunc = Callable[[DataType], DataType]
def query(data: DataType, selector: ModifierFunc,
*filters: ModifierFunc) -> DataType:
"""Query data with column selection and filters
:param data: List of dictionaries with columns and values
:param selector: result of `select` function call
:param filters: Any number of results of `field_filter` function calls
:return: Filtered data"""
pass
def select(*columns: str) -> ModifierFunc:
"""Return function that selects only specific columns from dataset"""
pass
def field_filter(column: str, *values: Any) -> ModifierFunc:
"""Return function that filters specific column to be one of `values`"""
pass
Хотя вопрос подразумевает, что select
и field_filter
могут быть классами, я не думаю, что здесь это необходимо; Я бы просто заставил их возвращать обычные старые кортежи:
select = field_filter = lambda *args: args
а затем query
— это просто понимание списка и словаря, когда вы перебираете список словарей и возвращаете поля select
ed из словарей, которые соответствуют field_filter
s:
def query(data, keys, *filters):
return [
{k: d[k] for k in keys}
for d in data
if all(d[k] in v for k, *v in filters)
]
friends = [
{'name': 'Sam', 'gender': 'male', 'sport': 'Basketball'},
{'name': 'Emily', 'gender': 'female', 'sport': 'volleyball'},
]
result = query(
friends,
select('name', 'gender', 'sport'),
field_filter('sport', *('Basketball', 'volleyball')),
field_filter('gender', *('male',))
)
print(result) # [{'name': 'Sam', 'gender': 'male', 'sport': 'Basketball'}]