学堂 学堂 学堂公众号手机端

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

lewis 1年前 (2024-04-03) 阅读数 5 #技术

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

示例代码如下:

importpandasaspd #读取整个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错误。因此,确保列名的正确性非常重要。

版权声明

本文仅代表作者观点,不代表博信信息网立场。

热门