的Python和熊貓：查找列表中的字典根據鍵值

Question

我有一個列表，它看起來像這樣：

dict_list = [{'angle': 5.0, 
       'basic_info': '111', 
       'x': [1,2,3,4], 
       'y': [3,2,1,4],}, 
      {'angle': 25.0, 
       'basic_info': '111', 
       'x': [1,2,3,4], 
       'y': [3,1,5,2],}, 
      {'angle': 3.0, 
       'basic_info': '111', 
       'x': [1,2,3,4], 
       'y': [1,1,4,1],},] 

我想要得到的字典角度25，我該怎麼辦呢？

---

UPDATE：

玩熊貓一段時間後，我覺得這可能是可能

df = pd.DataFrame(dict_list) 
temp = df.query("(angle ==25)").T.to_dict()[temp.keys()[0]] 
temp 

返回

{'angle': 25.0, 'basic_info': '111', 'x': [1, 2, 3, 4], 'y': [3, 1, 5, 2]} 

但是，這是一個有點劈。

Answer 1

假設在你的字典裏每一個角度都是獨特的，每個字典包含鍵「角」：

df = None 
for sub_dict in dict_list: 
    if sub_dict['angle'] == 25: 
     df = pd.DataFrame({'x': sub_dict['x'], 'y': sub_dict['y']}) 
     break # Stops after finding the first matching angle. 
if df is not None: 
    df.plot(x='x', y='y') 

Answer 2

內置filter功能會做你想要什麼......你剛剛需要爲它提供一個決定保留哪些對象的函數。下面的功能是has_angle。

我在filter_by_angle中包裝了has_angle，所以我可以使用functools.partial來避免硬編碼尋找的角度。 filter_by_angle是一個發電機功能，所以它可以yield任意數量的匹配形狀。

import functools 


def has_angle(angle, shape): 
    # get with default None prevents a KeyError on shapes 
    # without angles. 
    return shape.get('angle', None) == angle 


def filter_by_angle(angle, shapes): 
    filter_key = functools.partial(has_angle, angle) 
    yield from filter(filter_key, shapes) 


def main(): 
    dict_list = ... # Same as yours. 
    matching_shapes = filter_by_angle(25, dict_list) 
    for shape in matching_shapes: 
     print(shape) 
    return 

此打印：

{'y': [3, 1, 5, 2], 'angle': 25.0, 'x': [1, 2, 3, 4], 'basic_info': '111'} 

注意，yield from語法需要Python 3.3或更高版本。