使用以下方式为每行的条件设置不同的标签:
data_new.loc[data_new['节假日'].isin(['周六', '周日']), 'is周末'] = 1
data_new.loc[~data_new['节假日'].isin(['周六', '周日']), 'is周末'] = 0
print(data_new)
- 在第一行中,使用
isin()
函数来检查每行的'节假日'
列是否为'周六'
或'周日'
,如果是,则将'is周末'
列的值设为 1。 - 在第二行中,使用
~
运算符对条件取反,检查每行的'节假日'
列是否不是'周六'
或'周日'
,如果是,则将'is周末'
列的值设为 0。