扫描二维码下载沐宇APP

沐宇

微信扫码使用沐宇小程序

沐宇

usecols如何配合pandas使用

扬州沐宇科技
2024-06-14 11:28:31
usecols

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

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

扫码添加客服微信