在 Pandas 中,可以使用条件判断和索引切片的方法来选取符合条件的行或列。
假设你有一个名为 df 的 DataFrame,想要删除其中某一列大于20的元素,可以按如下步骤进行操作:
import pandas as pd
# 创建示例 DataFrame
df = pd.DataFrame({'A': [10, 20, 30], 'B': [25, 5, 15], 'C': [30, 40, 50]})
# 打印原始 DataFrame
print("原始 DataFrame:")
print(df)
# 选取 B 列中大于20的元素
mask = df['B'] > 20
# 从 DataFrame 中删除符合条件的行
df = df[~mask]
# 打印删除后的 DataFrame
print("删除后的 DataFrame:")
print(df)
上述代码首先创建了一个示例 DataFrame,然后选取了其中 B 列中大于20的元素,并将结果保存在名为 mask 的布尔型数组中。接着使用 ~ 运算符对 mask 取反,得到一个新的布尔型数组,表示需要保留的行。最后使用这个数组对 DataFrame 进行索引切片,得到删除了符合条件的行的新 DataFrame。
输出结果如下:
原始 DataFrame:
A B C
0 10 25 30
1 20 5 40
2 30 15 50
删除后的 DataFrame:
A B C
1 20 5 40
2 30 15 50