删除列表中项目的某些实例?

2024-04-24 22:57:22 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在编写一个代码来计算创建一个列表中的其他乐高积木(例如2x2=21x1)所需的乐高积木的数量,我已经完成了这部分工作。然而,我还想返回一个新的列表后,这个'组合'的过程,删除乐高作品,要么使用或组合(因此,不再可用)。我现在的代码剥离了乐高的所有实例,而不仅仅是那些“已使用”的实例。我想知道是否有办法从一个列表中删除某个项目乘以一个数字。提前谢谢。你知道吗

import hw4_util
legos = hw4_util.read_legos()

print 'Current legos',legos

def lego_work(lego,q,leglist): 
        count=0
        num1x1= leglist.count('1x1')
        num2x1= leglist.count('2x1')
        num2x2= leglist.count('2x2')
        num4x2= leglist.count('4x2')
        if lego =='1x1':
                count = num1x1
        elif lego == '2x1':
                if num2x1 == 0:
                        count = (num1x1/2)
                        leglist2= filter(lambda a: a != '1x1', leglist)
                else:
                        count= num2x1
                        leglist2=filter(lambda a: a != '2x1', leglist)
        elif lego =='2x2':
                if num2x2==0 and (num2x1/2) ==0:
                        count = (num1x1/4)
                        leglist2=  filter(lambda a: a != '1x1', leglist)
                elif num2x2==0 and num2x1>0:
                        count= num2x1
                        leglist2= filter(lambda a: a != '2x1', leglist)
                else:
                        count=num2x2
                        leglist2= filter(lambda a: a != '2x2', leglist)
        elif lego== '4x2':
                if num4x2==0 and (num2x2/2) and (num2x1/4)==0:
                        count = (num1x1/8)
                        leglist2= filter(lambda a: a != '1x1', leglist)
                elif num4x2==0 and num2x2==0 and num2x1>0:
                        count= num2x1
                        leglist2= filter(lambda a: a != '2x1', leglist)
                elif num4x2==0 and num2x2>0:
                        count=num2x2  
                        leglist2= filter(lambda a: a != '2x2', leglist)
                else:
                        count=num4x2
                        leglist2= filter(lambda a: a != '4x2', leglist)
        if count>=q:
                print 'You can use',q,lego,'legos for this.'
        elif count<q:
                print  " I don't have",q,lego,'legos.' 
        print 
        print 'Current legos:',leglist2
legotype= raw_input('Type of lego wanted =>')
print legotype
quantity= input('Quantity wanted =>')
print quantity

print lego_work(legotype,quantity,legos)

Tags: andlambdaifcountfilterprintlegoselif