如何在python中将4个列表合并为1

Question

我的代码是这样的

testGraph=open(input("Enter a file name:"))
for line in testGraph:
  temp=line.split()

输出结果是这样的

['0', '1']
['2', '1']
['0', '2']
['1', '3']

我想把它们变成

[['0', '1'],['2', '1'],['0', '2'],['1', '3']]

有人能帮帮我吗？

Answer 1

您可以使用itertools.chain()将列表展开为一个列表。

>>> cmd = ['ls', '/tmp']
>>> numbers = range(3)
>>> itertools.chain(cmd, numbers, ['foo', 'bar'])
<itertools.chain object at 0x0000000003943F28>
>>> list(itertools.chain(cmd, numbers, ['foo', 'bar']))
['ls', '/tmp', 0, 1, 2, 'foo', 'bar']

但是，如果您想要一个列表列表，那么添加到列表中将是您的选择。

testGraph=open(input("Enter a file name:"))
result = []
for line in testGraph:
  result.append(line.split())

Answer 2

List comprehension，为你工作

testGraph = open(input("Enter a file name:")) 
result = [line.split() for line in testGraph]

执行：

In [29]: testGraph = open('abc.txt')

In [30]: [line.split() for line in testGraph]
Out[30]: [['0', '1'], ['2', '1'], ['0', '2'], ['1', '3']]

Answer 3

这应该只需对您的代码进行最小的更改即可工作。然后，temp将拥有您想要的组合列表。

testGraph=open(input("Enter a file name:"))
temp = []
for line in testGraph:
  temp.append(line.split())