Копирование определенных c столбцов из текстового файла в новый текстовый файл с python

Question

Я новичок в Python языке.

У меня есть текстовый файл с несколькими разделенными пробелами столбцами, который выглядит так:

year month lat lon chl_a TSM NO2
1999 3.0 9.02 38.87 4.32 NaN NaN
1999 4.0 9.02 38.87 2.65 17.4 NaN 

Я хотел бы выбрать некоторые столбцы, желательно, позвонив по имени столбца (это имена на первый необработанный) и скопируйте выбранные столбцы в новый текстовый файл. Есть простой способ сделать это? Большое спасибо.

Answer 1

Можно использовать Pandas:

df = pd.read_csv('file.txt', sep='\s')

interested_cols = ['year', 'month','lat','lon']

df[interested_cols].to_csv('out.txt', sep=' ')