Python之对应【一个列表和另一个列表对应!超好用】(Zip函数)
对应
这次介绍一下词语和数字的对应方法(只要是对应 什么都可以!)
def read_category():
"""读取分类目录,固定 将分类目录转换为{类别:id}"""
categories = ['ATB', 'ATC', 'ATO', 'CCTV', 'SIV', 'TMS', '车门', '车载信号系统',
'电子地图', '广播', '空调', '牵引系统', '网络通讯', '制动系统', 'ATP']
categories = [x for x in categories]
cat_to_id = dict(zip(categories, range(len(categories)))) # 将得到的列表与正整数数组进行对应并构成词典
return categories, cat_to_id
zip
该函数用于将可迭代的对象作为参数,将对象中对应的元素打包成一个个元组,然后返回由这些元组成的列表。如果各个迭代器的元素个数不一致,则返回列表长度与最短的对象相同,利用*操作符,可以将元组解压为列表
a = [1, 2, 3] b = [4, 5, 6] c = [4, 5, 6, 7, 8] zipped = zip(a, b) # 打包为元组的列表 上下对应,两两一组 zip(a, c) # 元素个数与最短的列表一致 zip(*zipped) # 解压 返回二维矩阵格式 # 可以用于对二维列表取列 matrix = [[1,2,3],[4,5,6],[7,8,9]] print(zip(*matrix)) # [(1, 4, 7), (2, 5, 8), (3, 6, 9)]
这个函数真的好用!!码住码住
加个关键词:对应