Python如何从文本文件中读取指定列(仅提取ColA列)
解决方法
有两种简单方式可以实现只提取ColA列的需求:
方法一:读取时直接指定要加载的列(推荐)
在pd.read_csv()中使用usecols参数,仅指定ColA列。这种方式在读取阶段就只加载目标列,更节省内存、效率更高:
import pandas as pd df = pd.read_csv('New Text Document.txt', sep="|", usecols=['ColA']) print(df)
执行后输出:
ColA 0 3 1 6
方法二:读取全量数据后筛选目标列
如果已经读取了所有列,也可以直接通过列名筛选出ColA。使用双层方括号可以保持结果为DataFrame格式(和你期望的输出结构一致):
import pandas as pd df = pd.read_csv('New Text Document.txt', sep="|") df_colA = df[['ColA']] print(df_colA)
运行后同样会得到你需要的输出。
内容的提问来源于stack exchange,提问作者manu p
相关产品推荐
相关产品推荐

