Python выбрать несколько столбцов из текстового файла, разделенного табуляцией

Question

Я пытаюсь извлечь несколько столбцов из текстового файла и в конечном итоге с кодом ниже, который извлекает все столбцы от 17 до последнего. Я пытаюсь выяснить комбинацию регулярных выражений, чтобы сообщить коду, какие еще точные столбцы нужно получить, но я немного застрял.

Может кто-нибудь дать мне подсказку?

f = open("datafile1.txt", "r")
g = open("datafile2.txt", "w")

for line in f:
  if line.strip():
    g.write("\t".join(line.split("\t")[17:]))

f.close()
g.close()

Answer 1

Вы можете достичь этого с помощью простого понимания списка, например:

f = open("datafile1.txt", "r")
g = open("datafile2.txt", "w")

for line in f:
  if line.strip():
    g.write("\t".join([line.split("\t")[i] for i in [18, 26, 41, 45]]))

f.close()
g.close()