假设我有一个清单
['x1_0','x2_1','x3_0']
如何将上面的列表拆分为两个列表,以便第一个列表包含
['x1','x2','x3']
和第二个列表[0,1,0]
?
即。
('x1_0')
/ \
/ \
/ \
1st list 2nd list
'x1' 0
随意使用尽可能多的工具。这显然可以在一个for循环中完成,我知道。有一个更好的方法吗 ?。 使用列表理解的东西? ALS
答案 0 :(得分:6)
您可以使用zip
和列表理解:
>>> zip(*[i.split('_') for i in l])
[('x1', 'x2', 'x3'), ('0', '1', '0')]
如果你想将第二个元组的元素转换为int,你可以使用下面的嵌套列表理解:
>>> [[int(i) if i.isdigit() else i for i in tup] for tup in zip(*[i.split('_') for i in l])]
[['x1', 'x2', 'x3'], [0, 1, 0]]
前面的方法是执行此任务的正确方法,但正如您在评论中所说,作为较小的解决方案,您可以使用map
:
>>> l=[l[0],map(int,l[1])]
>>> l
[('x1', 'x2', 'x3'), [0, 1, 0]]
答案 1 :(得分:1)
k=["x1_0","x2_1","x3_0"]
k1=[x.split("_")[0] for x in k]
k2=[int(x.split("_")[1]) for x in k]
你可以这样做。