список понимания выбранных индексы

Question

Допустим, у меня есть файл, как это (вид, что были матрица):

abcde 
fghik 
lmnop 

Я хочу, чтобы поместить это в 2d списка, но только с колоннами до индекса 3:

# 0 1 2 3 
[['a','b','c','d'], 
    ['f','g','h','i'], 
    ['l','m','n','o']] 

Как это сделать, используя понимание списка? Я знаю, что могу зациклиться, но я ищу более чистый способ.

f = open('file.txt') 
lines = f.readlines() 
matrix = [[a for a in b] for b in lines] # this gets all columns, up to 4 

Я мог бы также использовать перечисление/если во внутреннем списке понимать, что нужно для столбца. Это самое чистое?

Answer 1

Если я правильно понял вопрос, это должно работать (но, возможно, я уже упрощаю). Обратите внимание, что [: 4] во внутреннем понимании:

f = open('file.txt') 
lines = f.readlines() 
matrix = [[a for a in b[:4]] for b in lines] # this gets all columns, up to 4 

Answer 2

Кусочка каждая строка до 3-я колонка: matrix = [[a for a in b][:3] for b in lines]

Answer 3

«но только с колоннами до 3»? Что это значит? Основываясь на вашем примере, вы имеете в виду «только первые четыре столбца»? Если это так, просто:

with f as open('file.txt'): 
    matrix = [list(line[:4]) for line in f] 

Вызов readlines ненужно (хуже, вредно, как он потребляет много памяти для больших файлов), вы можете перебирать строки просто файл.