温馨提示×

usecols如何配合pandas使用

小樊
109
2024-06-14 11:28:31
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

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

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

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读:Matplotlib怎么与Pandas库配合使用

0