您可以使用
boolean indexing:
df = pd.Dataframe({'Sales':[10,20,30,40,50], 'A':[3,4,7,6,1]})print (df) A Sales0 3 101 4 202 7 303 6 404 1 50s = 30df1 = df[df['Sales'] >= s]print (df1) A Sales2 7 303 6 404 1 50df2 = df[df['Sales'] < s]print (df2) A Sales0 3 101 4 20也可以
mask通过
~以下方式反转:
mask = df['Sales'] >= sdf1 = df[mask]df2 = df[~mask]print (df1) A Sales2 7 303 6 404 1 50print (df2) A Sales0 3 101 4 20
print (mask)0 False1 False2 True3 True4 TrueName: Sales, dtype: boolprint (~mask)0 True1 True2 False3 False4 FalseName: Sales, dtype: bool



