from lookup import Dataset, AggregatedDataset
ds1 = Dataset({'a': 1, 'b': 2, 'c': 3})
ds2 = Dataset({'a': 4, 'b': 5, 'c': 6})
ads = AggregatedDataset([ds1, ds2])
result = ads.lookup('a', 'ds2')
print(result) # 4
这个方法允许我们从每个数据集中检索给定键的值,并将结果合并为单个字典。这里的'ds2'参数表示要在哪个数据集中查找值。
result = ads.lookup('b')
print(result) # {'ds1': 2, 'ds2': 5}
如果不指定数据集的名称,则lookup方法将在所有聚合数据集中查找键及其对应的值,并将结果以字典形式返回。