pandas怎么读取csv指定列
扬州沐宇科技
2024-02-02 16:58:37
Pandas
使用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错误。因此，确保列名的正确性非常重要。
下一条：如何选择java xml的解析库
