使用索引操作符创建Python Pandas的子集DataFrame

索引操作符是用于创建子集数据框的方括号。首先让我们创建一个Pandas DataFrame。我们在数据框中有3列。

dataFrame = pd.DataFrame({"Product": ["智能电视", "ChromeCast", "扬声器", "耳机"],"开盘库存": [300, 700, 1200, 1500],"收盘库存": [200, 500, 1000, 900]})

用单个列创建子集：

dataFrame[['Product']]

用多列创建子集：

dataFrame[['开盘库存', '收盘库存']]

更多Pandas文章，请阅读：Pandas教程

示例

以下是完整的代码：

import pandas as pd

dataFrame = pd.DataFrame({"Product": ["智能电视", "ChromeCast", "扬声器", "耳机"],"开盘库存": [300, 700, 1200, 1500],"收盘库存": [200, 500, 1000, 900]})

print "DataFrame...\n",dataFrame

print"\n使用索引操作符显示子集：\n",dataFrame[['Product']]

print"\n显示多个列的子集：\n",dataFrame[['开盘库存','收盘库存']]

输出

这将产生以下输出：

DataFrame...
   开盘库存   收盘库存   Product
0          300             200     智能电视
1          700             500     ChromeCast
2         1200             1000    扬声器
3         1500             900     耳机

使用索引操作符显示子集：
       Product
0      智能电视
1     ChromeCast
2      扬声器
3       耳机

显示多个列的子集：
   开盘库存   收盘库存
0       300             200
1       700             500
2     1200             1000
3    1500             900