Python Pandas – 根据索引从列中选择特定值创建子集

如果要根据索引从列中选择特定值创建子集，请使用iloc()方法。首先导入pandas库

import pandas as pd

创建一个带有Product记录的Pandas DataFrame。我们有3个列

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

使用iloc()创建一个包含2列和前2行的子集

print"\n使用iloc()显示子集 = \n",dataFrame.iloc[0:2,0:2]

例子

以下是完整代码

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"\n显示子集:\n",dataFrame['Product']

# 使用iloc()创建一个包含2列和前2行的子集
print"\n使用iloc()显示子集 = \n",dataFrame.iloc[0:2,0:2]

输出结果

这将产生以下输出

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

显示子集:
0      SmartTV
1      ChromeCast
2      Speaker
3      Earphone
Name: Product, dtype: object

使用iloc()显示子集 =
   Closing_Stock   Opening_Stock
0           200            300
1           500            700