要比较某一列的值与一个变量,并创建新的列,可以使用pandas库中的apply方法。下面是示例代码:
import pandas as pd
df = pd.DataFrame({'score': [75, 80, 90, 85, 95]})
threshold = 85
df['pass'] = df['score'].apply(lambda x: 1 if x >= threshold else 0)
print(df)
输出结果为: score pass 0 75 0 1 80 0 2 90 1 3 85 1 4 95 1
从以上代码可以看出,我们通过apply方法将score列中的每个值和threshold进行比较,满足条件的值对应新列pass中的值为1,不满足条件的值对应新列pass中的值为0。最后输出DataFrame即可。