Python Pandas – 根据列名对DataFrame进行子集操作

要创建一个按列名子集的DataFrame，请使用方括号。像这样使用带有方括号（索引运算符）和特定列名的DataFrame-

dataFrame[‘列名’]

首先，使用别名导入所需的库-

import pandas as pd

使用Product记录创建一个Pandas DataFrame-

dataFrame = pd.DataFrame({"Product": ["SmartTV", "ChromeCast", "Speaker", "Earphone"],
"Opening_Stock": [300, 700, 1200, 1500], "Closing_Stock": [200, 500, 1000, 900]})

让我们获取子集，即我们只获取Product列记录-

dataFrame['Product']

示例

以下是代码

import pandas as pd

dataFrame = pd.DataFrame({"Product": ["SmartTV", "ChromeCast", "Speaker", "Earphone"],"Opening_Stock": [300, 700, 1200, 1500],"Closing_Stock": [200, 500, 1000, 900]})

print"DataFrame...\n",dataFrame

print"\nDisplaying a subset:\n",dataFrame['Product']

输出

这将生成以下输出-

DataFrame...
   Closing_Stock   Opening_Stock   Product
0       200            300         SmartTV
1       500            700         ChromeCast
2       1000           1200        Speaker
3       900            1500        Earphone

Displaying a subset:
0      SmartTV
1      ChromeCast
2      Speaker
3      Earphone
Name: Product, dtype: object