在一q群遇到一问题:
a = [['a','b'],['c','d','f']]
b = [['1','2',],['3','4','5']]
把上式转化为:
data = [[{'a':'1'},{'b':'2'}],[{'c':'3'},{'d':'4'},{'f':'5'}]]
比较简洁的方法是使用推导式和zip函数:
oo = [[{i:j} for i,j in zip(x,y)] for x,y in zip(a,b)]
#[[{'a':'1'},{'b':'2'}],[{'c':'3'},{'d':'4'},{'f':'5'}]]
代码拆分
zip函数用于将可迭代对象中的对应参数打包成元组。在这里外面一层推导式就是将a,b内的list对应,即:
oo = [(x,y) for x,y in zip(a,b)]
print(oo)
#[(['a', 'b'], ['1', '2']), (['c', 'd', 'f'], ['3', '4', '5'])]
里面一层推导式也及对单个元组内的两个list操作将其转化为字典:
x = ['a', 'b']
y = ['1', '2']
u= [{i:j} for i,j in zip(x,y)]
print(u)
#[{'a': '1'}, {'b': '2'}]
就这样经过两成循环,变成最终那样了~~
itertools库
虽然得出来了,还觉得代码简单,但是没得出前,还是费了些时间和遇到了一些坑。一开始没注意要转化为什么样,我直接将它转化为一个大字典了==。
c = zip(chain(*a),chain(*b))
c = zip(chain(*a),chain(*b))
d = dict(c)
#{'a': '1', 'b': '2', 'c': '3', 'd': '4', 'f': '5'}
这不符合~~条件。
chain是itertools模块的一个函数,用于将多个可迭代对象合成一个,符号 * 用于在函数中拆包用的,*args用于函数中接受多个参数:
a = ['a','b']
b = ['c','d','f']
#list(chain(a,b))
['a', 'b', 'c', 'd', 'f']
list(chain('123','word','hello'))
#['1', '2', '3', 'w', 'o', 'r', 'd', 'h', 'e', 'l', 'l', 'o']
#拆包
a = [['a','b'],['c','d','f']]
def pr(*x):
print(x)#(['a', 'b'], ['c', 'd', 'f'])
print(*x)#['a', 'b'] ['c', 'd', 'f'],相当于a1,a2 = x
a1,a2 = x
print(a1)#['a', 'b']
print(a2)#['c', 'd', 'f']
pr(*a)
也曾尝试过使用map函数,但是我觉得过于复杂化了。而且我最后得出的那个代码也有bug...就不贴代码了。
小总结
遇到的坑,是我对 dict 这个构建字典的函数产生误解了。也是我一直以来没对这dict认真了解过,自以为是的以为:
dict(a,b)
#以为可以得出{a:b} ,囧~
#我help(dict),发现没有这用法·····
dict(one='1',two='2')
#{'one':'1','two':'2'}这样可以。。。。。
以前虽然也知道python的itertools库有许多有用的函数,但也没细看,今天到是好好看了下(算是) 。官方文档的确是个好东西,看pytorch时也发现了,奈何英语硬伤。
以前都没有记录问题的习惯,往往同一个问题,不多就又忘记怎么解决了。来简书记录平常遇到的问题,养成好习惯。