简单学量化pandas的应用32读取csv文件6

今天，我们学习read_csv函数的参数usecols。这个参数看字面意思就可以明白，是用来设置导入哪些列的。如果一张表有很多列，而我们只是要用到其中某些列，那就没有必要导入所有的列，所以在导入时就需要用usecols这个参数。这个参数的用法为：usecols=[列名称或列序号]。

下面我们举例使用一下，首先导入数据表：

import pandas as pd
df=pd.read_csv("stock.csv")
df

返回：

这是一个包含了交易日期、开高低收及成交量的数据表。现在我们不想显示开盘、最高及最低价格，那就这样导入：

import pandas as pd
df=pd.read_csv("stock.csv",usecols=['ts_code','trade_date','close','volume'])
df

返回：

将usecols那儿改为usecols=[0,1,5,6]也能实现一样的效果（即用列的位置替换列的名称）。