新闻资讯

质量为本、客户为根、勇于拼搏、务实创新

< 返回新闻资讯列表

pandas如何读取csv指定列,pandas 读取 csv

发布时间:2024-02-02 16:58:37

pandas如何读取csv指定列

使用pandas的read_csv函数读取csv文件,并通过指定columns参数来选择需要读取的列。

示例代码以下:

import pandas as pd

# 读取全部csv文件,不指定列
df = pd.read_csv('data.csv')

# 读取指定列
selected_columns = ['column1', 'column2', 'column3']
df_selected = pd.read_csv('data.csv', usecols=selected_columns)

在第二个例子中,使用了usecols参数来指定需要读取的列,将列名以列表情势传递给该参数。

需要注意的是,如果要读取的csv文件中的列名与selected_columns中的列名不完全匹配,可能会出现KeyError毛病。因此,确保列名的正确性非常重要。