以下是一个使用Python Pandas库比较两列并过滤相邻班级的列的示例代码:
import pandas as pd
# 创建示例数据
data = {'班级': ['1班', '2班', '3班', '4班', '5班'],
'人数': [30, 25, 28, 32, 29]}
df = pd.DataFrame(data)
# 添加相邻班级列
df['相邻班级'] = df['班级'].shift(-1)
# 过滤相邻班级的列
filtered_df = df[df['班级'] != df['相邻班级']]
# 打印结果
print(filtered_df)
运行以上代码,输出结果为:
班级 人数 相邻班级
0 1班 30 2班
1 2班 25 3班
2 3班 28 4班
3 4班 32 5班
4 5班 29 NaN
在示例代码中,我们首先创建了一个包含班级和人数的DataFrame。然后,我们使用shift
函数在DataFrame中添加了一个新的列相邻班级
,该列将原始班级列向上移动一行。最后,我们使用条件过滤筛选出班级与相邻班级不相同的行,即过滤掉相邻班级的列。