我有一個數(shù)據(jù)集（在一個文件中），它由具有各種屬性（列）的多個觀察值（行）組成。例如：AttrA AttrB AttrC 1 12 'a' 2 43 'd' 3 23 'f' 4 25 'z' 我將此數(shù)據(jù)集放入python字典中，使得：data = {'AttrA':[1,2,3,4],'AttrB':[12,43,23,25],'AttrC':['a','d','f','z']}我希望能夠基于其中一個鍵的標(biāo)準(zhǔn)來過濾觀察值。舉些例子。過濾對AttrA> = 3的觀察，例如： AttrA AttrB AttrC 3 23 'f' 4 25 'z' 或者reducedData = {'AttrA':[3,4],'AttrB':[23,25],'AttrC':['f','z']}看來您可以執(zhí)行以下操作：reduceddata = {(k,v) for k,v in data if (??)}，但是我不確定if語句之后會發(fā)生什么。另外，字典是否是此示例中使用的最佳數(shù)據(jù)類型？如果數(shù)據(jù)在嵌套列表中，則似乎更容易過濾。
3 回答
素胚勾勒不出你
TA貢獻(xiàn)1827條經(jīng)驗 獲得超9個贊
我將首先更改結(jié)構(gòu)：
table = [dict(zip(data.keys(), row)) for row in zip(*data.values())]
它看起來像這樣：
[{'AttrA': 1, 'AttrB': 12, 'AttrC': 'a'},
{'AttrA': 2, 'AttrB': 43, 'AttrC': 'd'},
{'AttrA': 3, 'AttrB': 23, 'AttrC': 'f'},
{'AttrA': 4, 'AttrB': 25, 'AttrC': 'z'}]
現(xiàn)在，您可以完全按照您的描述進(jìn)行過濾：
[row for row in table if row['AttrA'] >= 3]
拉風(fēng)的咖菲貓
TA貢獻(xiàn)1995條經(jīng)驗 獲得超2個贊
def my_filter(data,attr,val):
ind = [i for i,x in enumerate(data[attr]) if x >= val]
reducedData = {k: [v[i] for i in ind] for k,v in data.items()}
return reducedData
data = {'AttrA':[1,2,3,4],'AttrB':[12,43,23,25],'AttrC':['a','d','f','z']}
print my_filter(data, 'AttrA',3)
輸出：
{'AttrB': [23, 25], 'AttrC': ['f', 'z'], 'AttrA': [3, 4]}
DIEA
TA貢獻(xiàn)1820條經(jīng)驗 獲得超3個贊
我想我會選擇：
data = [(12, 'a'), (43, 'd'), (23, 'f'), (25, 'z')]
data_dic = dict(enumerate(data, 1))
reducedData = {k:v for k, v in data_dic if k >= 3}
添加回答
舉報
0/150
提交
取消