Python :删除数据帧的行并保留特定的组

import pandas as pd

# 创建一个示例数据帧
data = {
    'group': ['A', 'A', 'B', 'B', 'C', 'C'],
    'value': [10, 20, 30, 40, 50, 60]
}
df = pd.DataFrame(data)

# 打印原始数据帧
print("原始数据帧:")
print(df)

# 假设我们想要保留group为'A'的所有行
group_to_keep = 'A'

# 使用条件筛选保留特定的组
df_filtered = df[df['group'] == group_to_keep]

# 打印筛选后的数据帧
print("\n保留特定组后的数据帧:")
print(df_filtered)

原始数据帧:
  group  value
0     A     10
1     A     20
2     B     30
3     B     40
4     C     50
5     C     60

保留特定组后的数据帧:
  group  value
0     A     10
1     A     20

# 假设我们想要删除group为'B'的所有行
group_to_remove = 'B'

# 使用drop方法删除特定的组
df_dropped = df.drop(df[df['group'] == group_to_remove].index)

# 打印删除特定组后的数据帧
print("\n删除特定组后的数据帧:")
print(df_dropped)

删除特定组后的数据帧:
  group  value
0     A     10
1     A     20
4     C     50
5     C     60