包含列表數據的字典，基於列表中的值的過濾器

Question

我有基於多個輸入收集的測試數據，並生成單個輸出。我正在將這些數據存儲在一個字典中，其中的鍵是我的參數/結果標籤，其值是測試條件和結果。我希望能夠過濾數據，以便可以基於孤立的條件生成圖。

在我的例子中，我的測試條件是'a'和'b'，實驗的結果是'c'。我想過濾我的數據，所以我得到一個具有相同關鍵字，值結構的字典，只有我的過濾結果。然而，我目前的字典理解返回一個空的字典。任何建議，以獲得理想的結果？

目前代碼：

data = {'a': [0, 1, 2, 0, 1, 2], 'b': [10, 10, 10, 20, 20, 20], 'c': [1.3, 1.9, 2.3, 2.3, 2.9, 3.4]} 
filtered_data = {k:v for k,v in data.iteritems() if v in data['b'] >= 20} 

所需的結果：

{'a': [0, 1, 2], 'b': [20, 20, 20], 'c': [2.3, 2.9, 3.4]} 

當前結果：

{} 

而且，是本詞典列出的一個很好的模式來存儲這類數據，因爲我要過濾結果，還是有更好的方法來實現這一目標？

Answer 1

使用本：

k:[v[i] for i,x in enumerate(v) if data['b'][i] >= 20] for k,v in data.items()} 

所需的結果：

{'a': [0, 1, 2], 'c': [2.3, 2.9, 3.4], 'b': [20, 20, 20]} 

Answer 2

考慮在這類工作中使用熊貓模塊。

import pandas as pd 
df = pd.DataFrame(data) 
df = df[df["b"] >= 20] 
print(df) 

看起來像這樣會給你你想要的。您正在使用字典鍵來表示列名稱，並且這些值只是給定列中的行，因此它可以使用數據框。

結果：

a b c 
3 0 20 2.3 
4 1 20 2.9 
5 2 20 3.4 

Answer 3

是否所有匹配訂單字典值列表？如果是這樣，你可以看看你想要過濾的列表，例如'b'，在這種情況下，找到你想要的值，然後在字典中的其他值上使用這些索引或相同的片。

例如：

matching_indices = [] 
for i in data['b']: 
    if data['b'][i] >= 20: 
     matching_indices.append(i) 
new_dict = {} 
for key in data: 
    for item in matching_indices: 
     new_dict[key] = data[key][item] 

你也許可以推測它的字典理解，如果你想要的。希望這很清楚。

Answer 4

你可以改變這種方法，它會給它更多的靈活性。您當前的邏輯意味着，數據集A和C被忽視，因爲沒有值大於或等於20：

data = {'a': [0, 1, 2, 0, 1, 2], 'b': [10, 10, 10, 20, 20, 20], 'c': [1.3, 1.9, 2.3, 2.3, 2.9, 3.4]} 
filter_vals = ['a', 'b'] 
new_d = {} 
for k, v in data.iteritems(): 
    if k in filter_vals: 
    new_d[k] = [i for i in v if i >= 20] 

print new_d 

現在我不是一個大風扇，如果很多if語句，但這樣的事情是直線前進並且可以調用多次

def my_filter(operator, condition, filter_vals, my_dict): 
    new_d = {} 
    for k, v in my_dict.iteritems(): 
    if k in filter_vals: 
     if operator == '>': 
     new_d[k] = [i for i in v if i > condition] 
     elif operator == '<': 
     new_d[k] = [i for i in v if i < condition] 
     elif operator == '<=': 
     new_d[k] = [i for i in v if i <= condition] 
     elif operator == '>=': 
     new_d[k] = [i for i in v if i >= condition] 
    return new_d 

Answer 5

我同意上面的熊貓方法。

如果由於某種原因，你討厭大熊貓或是一位老學校的計算機科學家，元組是一種很好的方法來關聯關係數據。在你的例子中，a，b和c列表是列而不是行。對於元組，你會希望保存爲行：

data = {'a':(0,10,1.3),'b':(1,10,1.9),'c':(2,10,2.3),'d':(0,20,2.3),'e':(1,20,2.9),'f':(2,20,3.4)} 

其中，元組存儲在（條件1，條件2，結果）格式，你描述的，你可以調用單個測試或過濾爲您介紹一套。從那裏，你可以得到一個過濾的結果如下：

filtered_data = {k:v for k,v in data.iteritems() if v[1]>=20} 

返回：

{'d': (0, 20, 2.3), 'e': (1, 20, 2.9), 'f': (2, 20, 3.4)}