以下是一个示例代码,用于判断数据帧中是否存在特定组ID,并在不存在时添加一行数据:
import pandas as pd
# 创建一个示例数据帧
df = pd.DataFrame({'GroupID': [1, 2, 3, 4],
'Value': [10, 20, 30, 40]})
# 要检查的组ID
search_group_id = 3
# 检查数据帧中是否存在特定组ID
if search_group_id in df['GroupID'].values:
print(f"GroupID {search_group_id} 存在于数据帧中")
else:
# 不存在时,在数据帧底部添加一行数据
new_row = {'GroupID': search_group_id, 'Value': 50}
df = df.append(new_row, ignore_index=True)
print(f"GroupID {search_group_id} 不存在于数据帧中,已添加一行数据")
# 输出更新后的数据帧
print(df)
运行结果如下所示:
GroupID 3 存在于数据帧中
GroupID Value
0 1 10
1 2 20
2 3 30
3 4 40
如果要检查的组ID不存在于数据帧中,将会在底部添加一行数据。
下一篇:按最高出现次数分组的SQL