python - 在Python中迭代对象实例的特定属性

Question

我是一名新手程序员，所以请原谅我没有使用 Python 特定的词汇。

假设我定义一个类 CarSpecs，其属性为 CarReg、Make、Model 和 Color，创建该类的多个实例(称为记录)并将它们一一附加到名为 SuperCars 的文本文件中。我希望我的程序做的是读取整个文件并返回红色汽车的数量(即通过查找每个实例的属性Color)。

这是我到目前为止所做的事情:

定义一个类:

class Carspecs(object):  
    def __init__(self, carreg, make, model, color):    
        self.CarReg = carreg   
        self.Make = make  
        self.Model = model   
        self.Color = color 

然后我创建了几个实例并定义了一个函数来将实例(或者你可以说“记录”)添加到SuperCars:

def addCar(CarRecord):  
   import pickle  
       CarFile = open('Supercars', 'ab')  
       pickle.dump(CarRecord, CarFile)  
       CarFile.close()  

接下来我该怎么做才能输出红色汽车的数量？

Answer 1

您必须再次打开该文件，读取所有记录，然后查看哪些汽车 Color 属性等于红色。因为您要将每个实例保存在 pickle 中，所以您必须执行如下操作:

>>> with open('Supercars', 'rb') as f:
...    data = []
...    while True:
...        try:
...            data.append(pickle.load(f))
...        except EOFError:
...            break
...
>>>
>>> print(x for x in data if x.Color == 'red')

我建议您将数据存储在一个列表中并腌制该列表，这样您就不必使用那个hacky循环来获取所有项目。存储这样的列表很容易。假设您已经创建了 CarSpec 对象列表并将它们存储在列表 records 中:

>>> with open('Supercars', 'wb') as f:
...    pickle.dump(records, f)
...
>>>

然后阅读它就像这样简单:

>>> with open('Supercars', 'rb') as f:
...     data = pickle.load(f)
...
>>>

您甚至可以轻松过滤它:

>>> with open('Supercars', 'rb') as f:
...     data = [x for x in pickle.load(f) if x.Color == 'Red']
...
>>>

如果您想在将它们存储在pickle中之前显示它们，您只需迭代记录列表并用红色打印汽车即可。