所以,我只是在尝试解决hackerrank.com上的一个问题
下面是我的代码:
if __name__ == '__main__':
N = int(input())
num_list = []
for numberOfCommands in range(N):
command , * numbers = input().split(" ")
numbers = [int(i) for i in numbers]
print(numbers)
if command == 'insert':
num_list.insert(numbers[0],numbers[1:])
elif command == 'print':
print(num_list)
else:
num_list.command(numbers[0])
问题是当我打印list
时,它会打印嵌套列表。我想要的是创建一个int
列表。
发布于 2018-06-03 01:14:14
您可以像这样使用flatten a list of lists:
flat_list = [element for sublist in num_list for element in sublist]
print(flat_list)
或者使用内置的itertools.chain
import itertools
list(itertools.chain.from_iterable(num_list))
https://stackoverflow.com/questions/50658713
复制相似问题