创建新的Dataframe数据切片问题

Question

这是我的代码片段。但是代码有效，我收到以下错误：

“正在尝试在DataFrame的切片副本上设置值”

我猜这是由于一些不推荐的语法...

new_data['mon_fri'] = 0
for i in range(0,len(new_data)):
    if (new_data['Dayofweek'][i] == 0 or new_data['Dayofweek'][i] == 4):
        new_data['mon_fri'][i] = 1
    else:
        new_data['mon_fri'][i] = 0

Answer 1

如果存在矢量化替代品，请勿在pandas中循环，这里可以使用isin作为布尔掩码并转换为True/False到1/0映射的整数：

new_data['mon_fri'] = new_data['Dayofweek'].isin([0,4]).astype(int)

或者使用numpy.where：

new_data['mon_fri'] = np.where(new_data['Dayofweek'].isin([0,4]), 1, 0)