python - 使用 Pandas 进行条件切片(优雅的方式)

Question

我需要根据某些标准拆分整个数据框 (Pandas)。例如:

import pandas as pd
import numpy as np

a = np.random.random(size=(1,10))
b = np.random.randint(100,size=a.shape)
df = pd.DataFrame(array((a,b)).T,columns=['a','b'])

现在，如果我想将数据框分成两部分，其中 df['a'] >= 30 和 df['a'] < 30

df_two = [df[df['a'] < 30], df[df['a']>=30]]

是否有更优雅的方法根据某些条件拆分数据框？使用列表理解或循环？例如，如果我有多个条件而不仅仅是一个条件，并且我想遍历条件列表而不是手动执行。

Answer 1

您可以使用:

mask = df['a'] < 30
df_two = [df[mask], df[~mask]]