温馨提示×

usecols如何配合pandas使用

小樊
108
2024-06-14 11:28:31
栏目: 编程语言

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

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

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

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

import pandas as pd

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

输出结果为:

   A  C
0  1  3
1  5  7

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

0