背景
写代码的时候,你会发现你的代码越写越多。
然而,功能需要也越来越多,然后你的冗余代码就多得不能再多了~~~怎么办,我太难了。
那就寻求一些高级写法,一般的高级写法都是尽可能地短。
另外,把重复的代码抽取出来,封装成函数,每次使用直接调函数即可。
For循环赋值
前提条件:我创建了一个road类,这个类里面有这些属性。我先初始化给road赋值,然后再把这些对象放到roadObjList集合里面。
目标:从roadObjList集合里面取出每个road对象的objectid值,然后放入到新的列表里面。
class road():
def __init__(self,gid,objectid,name,width,direction,speed,type,snode,enode,length,startX,startY,endX,endY):
self.gid= gid
self.objectid = objectid
self.name = name
self.width = width
self.direction = direction
self.speed = speed
self.type = type
self.snode = snode
self.enode = enode
self.length = length
self.startX = startX
self.startY = startY
self.endX = endX
self.endY = endY
写法一:
objList = []
for r in roadObjList:
objList.append(r.objectid)
写法二:
objList = [r.objectid for r in roadObjList]
补充
关于for循环遍历的写法,还有好几种,虽然写法很low,但是效果都是差不多的。
写法一:
for i,ele in enumerate(roadPointObjList):
print(i)
print(ele)
写法二:
for i in range(0, len(roadPointObjList)):
print(i)
print(roadPointObjList[i])
总结
我觉得两种写法其实都是一样的,时间复杂度都是O(n),毕竟都要遍历for循环。
只是第二种写法一步到位,当然如果要在for循环里面作判断的话,那就选第一种写法了。