как получить вход в определенном порядке из файла в Python

Question

У меня есть файл, содержащий множество тестовых случаев, и я хочу взять их в качестве входных данных, контейнер файла выглядит так:

1 232 4343
2 343 5454
3 545 6556
...

Я хочу иметь список карт, чтобы вводсохраниться так:

[[232,4343], [343, 5454], [545,6556] , ...]

первый вход (количество строк) легко получить, просто используя индекс списка в списке, но как я могу получить другие мои входы и сохранить их всписок списков?
Я использую python 3.6.5

Answer 1

Попробуйте это:

with open(filename,'r') as f:
    l=[list(map(int,i.rstrip().split()[1:])) for i in f]

А теперь:

print(l)

Это:

[[232,4343], [343, 5454], [545,6556]]

Или быстрее используйте Панд:

import pandas as pd
df=pd.read_csv(filename,sep='\s+',header=None,index_col=0)
print(df.values.tolist())

Вывод:

[[232, 4343], [343, 5454], [545, 6556]]

Обновление:

with open(filename,'r') as f:
    l=[list(map(int,i.rstrip().split())) for i in f]

Вывод:

[[1, 232, 4343], [2, 343, 5454], [3, 545, 6556]]

Или с пандами:

import pandas as pd
df=pd.read_csv(filename,sep='\s+',header=None)
print(df.values.tolist())

Для этого требуется меньше кода ...