将观测值表示为字典，并基于一个键和值集进行过滤

Question

我有一个数据集（在一个文件中），它由具有各种属性（列）的多个观察值（行）组成。例如：

AttrA   AttrB   AttrC

 1       12      'a'

 2       43      'd'

 3       23      'f'

 4       25      'z' 

我将此数据集放入python字典中，使得：

data = {'AttrA':[1,2,3,4],'AttrB':[12,43,23,25],'AttrC':['a','d','f','z']}

我希望能够基于其中一个键的标准来过滤观察值。举些例子。过滤对AttrA> = 3的观察，例如：

 AttrA   AttrB   AttrC

 3       23      'f'

 4       25      'z' 

或者

reducedData = {'AttrA':[3,4],'AttrB':[23,25],'AttrC':['f','z']}

看来您可以执行以下操作：reduceddata = {(k,v) for k,v in data if (??)}，但是我不确定if语句之后会发生什么。另外，字典是否是此示例中使用的最佳数据类型？如果数据在嵌套列表中，则似乎更容易过滤。

Answer 1

我将首先更改结构：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'}]现在，您可以完全按照您的描述进行过滤：[row for row in table if row['AttrA'] >= 3]

Answer 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]}

Answer 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}