仅基于一个键的值过滤所有字典

Question

我有这个字典：

d1 = {"date":["2019-11","2019-11","2019-12","2019-12","2020-01","2020-01"],
      "shares":[28,16,5,10,1,1],
      "target":["INSTAGRAM","TEXT","INSTAGRAM","TEXT","INSTAGRAM","TEXT"]
     }

我只想为d1过滤所有target = INSTAGRAM词典，而不使用pandas。所以这是预期的结果：

{"date":["2019-11","2019-12","2020-01"],
 "shares":[28,5,1],
 "target":["INSTAGRAM","INSTAGRAM","INSTAGRAM"]
}

Answer 1

使用defaultdict：

defaultdict

---

from collections import defaultdict
d = defaultdict(list)
for i, target in enumerate(d1['target']):
    if target == 'INSTAGRAM':
        for k,v in d1.items():
            d[k].append(v[i])

---

如果要避免任何导入并使用python的

print(d)
defaultdict(list,
            {'date': ['2019-11', '2019-12', '2020-01'],
             'shares': [28, 5, 1],
             'target': ['INSTAGRAM', 'INSTAGRAM', 'INSTAGRAM']})

类：

dict

Answer 2

这应该可以解决问题：

d = dict()
for i, target in enumerate(d1['target']):
    if target == 'INSTAGRAM':
        for k,v in d1.items():
            if k in d.keys():
                d[k].append(v[i])
            else:
                d[k] = [v[i]]

def filter_by_key(key, value, _dict):
    """ Filter a dictionary by it's key value

    :param str key: Dictionary key to inspect
    :param str value: Value to look for inside the dictionary key
    :param dict _dict: Dictionary to inspect
    :return: Dictionary with filter applied
    """

    indexes = [index for index, item in enumerate(_dict[key])
               if _dict[key][index] == value]

    result_dict = {}
    for key, value in a.iteritems():
        result_dict[key] = [_dict[key][index] for index in indexes]
    return result_dict