要按照条件从数据帧中选择行,可以使用布尔索引。下面是一个示例代码,演示如何按组选择数据帧中的行,基于条件:
import pandas as pd
# 创建示例数据帧
data = {'组': ['A', 'A', 'B', 'B', 'C', 'C'],
'数值': [1, 2, 3, 4, 5, 6]}
df = pd.DataFrame(data)
# 按组选择数值大于3的行
result = df[df['数值'] > 3]
print(result)
输出:
组 数值
3 B 4
4 C 5
5 C 6
在上面的示例中,我们首先创建了一个包含组和数值的数据帧。然后,我们使用布尔索引来选择数值大于3的行。最后,我们打印出结果数据帧。
注意,这里的df['数值'] > 3
返回的是一个布尔数组,其中对应的元素为True表示该行满足条件。因此,使用这个布尔数组作为索引,可以选择满足条件的行。
上一篇:按组选择n个数据并限制
下一篇:按组选择所有具有指定类的复选框