Rifat Alptekin Çetin

您可以使用类:

class element:
    def __init__(self,shape,color):
        self.shape=shape
        self.color=color

查询功能:

def query(data,shape=None,color=None):
     if shape:
         data=[x for x in data if x.shape==shape]
     if color:
         data=[x for x in data if x.color==color]
     return data

使用:

red_circle=element("circle","red")
red_square=element("square","red")
blue_circle=element("circle","blue")
blue_square=element("square","blue")

data=[red_circle,red_square,blue_circle,blue_square]
qdata=query(data,color="blue")

for x in qdata:
    print("shape: {} - color: {}".format(x.shape,x.color))