扫描二维码下载沐宇APP

沐宇

微信扫码使用沐宇小程序

沐宇

usecols如何避免读取无用列

扬州沐宇科技
2024-06-14 11:33:38
usecols

在使用usecols参数时,可以指定要读取的列的索引或列名,从而避免读取无用列。通过指定需要的列,可以减少内存和处理时间,提高读取数据的效率。

例如,假设有一个包含5列的CSV文件,但我们只需要读取第1列和第3列的数据,可以这样使用usecols参数:

import pandas as pd

# 读取文件并只选择需要的列
df = pd.read_csv('file.csv', usecols=[0, 2])

print(df)

在这个例子中,usecols=[0, 2]指定只读取第1列和第3列的数据,而忽略其他列。这样就避免了读取无用列,提高了数据读取的效率。

扫码添加客服微信