当前位置: 首页> 服务器> 正文

usecols如何配合pandas使用

usecols如何配合pandas使用

usecols 参数是 pandas.read_csv() 函数中的一个参数,用于指定读取 CSV 文件时要选择的列。该参数接受一个列表,其中包含要读取的列名或列索引。

例如,假设有一个 CSV 文件 data.csv 包含以下内容:

A,B,C,D 1,2,3,4 5,6,7,8

要读取只包含列 A 和 C 的数据,可以使用下面的代码:

import pandas as pd df = pd.read_csv('data.csv', usecols=['A', 'C']) print(df)

输出结果为:

A C 0 1 3 1 5 7

这样就只读取了指定的列,而不是全部列。