python - 使用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 

我想通过调用列名（这是第一个原始名称）来选择一些列，并将选定的列复制到一个新的文本文件中。有没有简单的方法来做到这一点？非常感谢你。

标签： pythonpandasnumpy

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=' ')