我想要一个公式或任何类似“开关”的东西。如果“位置”列达到3或以上,则开关打开(= 1)。如果“位置”高于5,则开关关闭(= 0)。如果位置低于3,则开关也将关闭(= 0)。我已将“所需”列包括在内,以显示我希望该新列自动化的内容。
df = pd.DataFrame()
df['position'] = [1,2,3,4,5,6,7,8,7,6,5,4,3,2,1,2,3,4,5,4,3,2,1]
df['desired'] = [0,0,1,1,1,0,0,0,0,0,0,0,0,0,0,0,1,1,1,1,1,0,0]
the_list = [1,2,3,4,5,6,7,8,7,6,5,4,3,2,1,2,3,4,5,4,3,2,1]
def check(number):
if number > 2:
return 1
else:
return 0
print(list(map(check,the_list)))