在pandas中,可以使用条件筛选的方式来只保留特定的值。以下是一种常见的方法:
import pandas as pd
# 创建一个示例DataFrame
df = pd.DataFrame({'A': [1, 2, 3, 4, 5],
'B': ['a', 'b', 'c', 'd', 'e']})
# 保留'A'列中值为2和4的行
df = df[df['A'].isin([2, 4])]
print(df)
输出结果为:
A B
1 2 b
3 4 d
在这个例子中,我们使用df['A'].isin([2, 4])
创建了一个布尔条件,该条件表示只保留'A'列中值为2和4的行。然后,我们将这个条件应用于DataFrame的索引中,以筛选出满足条件的行。
import pandas as pd
# 创建一个示例DataFrame
df = pd.DataFrame({'A': [1, 2, 3, 4, 5],
'B': ['a', 'b', 'c', 'd', 'e']})
# 保留'A'列中值为2和4的行
df = df.loc[df['A'].isin([2, 4])]
print(df)
输出结果为:
A B
1 2 b
3 4 d
在这个例子中,我们使用df['A'].isin([2, 4])
创建了一个布尔条件,该条件表示只保留'A'列中值为2和4的行。然后,我们使用loc方法将这个条件应用于DataFrame,以选择满足条件的行。
以上是在pandas中只保留特定值的两种常见方法。在实际应用中,可以根据具体需求选择适合的方法。
领取专属 10元无门槛券
手把手带您无忧上云