Python：使用Python将名称和数字仅从txt文件复制到Excel中

Question

我有一个巨大的文本文件，由包括学生的姓名和唯一ID的详细信息组成，如下所示。

所有学生的团体名单2020年6月5日BRD Rahul e34 Pradeep e44 Venkat r45 Azhar t54小组KRD Fairoz e45 Kumar e55 Akshay e44 Vivek e99

当我运行它时，我需要在excel / csv工作表中输出，其中仅显示名称（列1）和唯一ID（列2）。

这是我尝试过的

import pandas as pd
df = pd.read_csv('C:\Users\PMishra\Desktop\Document.txt', delimiter=',')
df.to_csv('C:\Users\PMishra\Desktop\Output.csv')

当我运行它时，它将只将所有内容从文本文件复制到Excel工作表。我想在仅显示Name（Column1）和Unique ID（Column2）的excel / csv工作表中输出。我是python（Spyder）的新手。请帮助。

Answer 1

使用参数usecols，您可以指定要加载.read_csv()的列；

import pandas

df = pandas.read_csv('C:\Users\PMishra\Desktop\Document.txt', delimiter=',',
                     usecols=[0, 1])

df.to_csv('C:\Users\PMishra\Desktop\Output.csv')

Answer 2

没有列名（使用列索引/编号）

import pandas as pd
columns_to_keep = ['Name', 'Unique id']
df = pd.read_csv('C:\Users\PMishra\Desktop\Document.txt', usecols=[0, 1], delimiter=',')
df.to_csv('C:\Users\PMishra\Desktop\Output.csv')

带有列名

import pandas as pd
columns_to_keep = ['Name', 'Unique id']
df = pd.read_csv('C:\Users\PMishra\Desktop\Document.txt', delimiter=',')
df = df[columns_to_keep]
df.to_csv('C:\Users\PMishra\Desktop\Output.csv')