鉴于以下案文:
input_value= """12
23
54"""
我想返回包含每行的第一个和第二个元素的列表:
[[1,2,5], [2,3,4]]
我试图使用一行程序来实现这一点,但是很难访问每一行的正确索引,然后将它们放入单独的列表中.返回的列表数量应等于每行上显示的位数:
我已经try 了以下作为基础,但显然需要更多的逻辑才能检索到正确的数字:
new_list = [[elem] for elem in input_value.split('\n')]
鉴于以下案文:
input_value= """12
23
54"""
我想返回包含每行的第一个和第二个元素的列表:
[[1,2,5], [2,3,4]]
我试图使用一行程序来实现这一点,但是很难访问每一行的正确索引,然后将它们放入单独的列表中.返回的列表数量应等于每行上显示的位数:
我已经try 了以下作为基础,但显然需要更多的逻辑才能检索到正确的数字:
new_list = [[elem] for elem in input_value.split('\n')]
使用嵌套的list comprehension:
out = [[int(i) for i in x] for x in zip(*input_value.split('\n'))]
输出:
[[1, 2, 5], [2, 3, 4]]