温馨提示×

usecols在读取csv文件中的应用

小樊
89
2024-06-14 11:35:37
栏目: 编程语言

usecols参数用于选择要读取的列。可以传入一个包含列名或列索引的列表,或者是一个函数来过滤列。这样可以在读取大型CSV文件时只选择需要的列,避免加载整个文件到内存中。

示例:

import pandas as pd

# 读取csv文件,只选择指定列
df = pd.read_csv('data.csv', usecols=['col1', 'col2'])

# 使用函数过滤列
df = pd.read_csv('data.csv', usecols=lambda x: 'col' in x)

0