reset_index(drop=True)

今天必须写个博文mark一下了,错了不知道几百次的弱智问题。

train_x = data[data.label != -1].reset_index(drop=True)#注意!否则报错
train_y = train_x.pop('label').values

train_index = train_x[train_x.A == 2018].index.tolist()
test_index = train_x[train_x.B == 1016].index.tolist()

print(train1_x[train_index], train1_y[train_index])
print(train1_x[test_index], train1_y[test_index])
#train1_x是对trainx的列进行某些操作得到的,进行过reset_index(drop=True),所以index连续。如:
print(train1_x.index.tolist()) #[1,2,3,4]

#而train_x = data[data.label != -1],如果不进行reset_index(drop=True),index会变得不连续。如:
print(train_x.index.tolist()) #[1,2,4,5]

#在后面需要对train1_x进行操作时,如果前面不对train_x进行reset_index(drop=True),那么train1_x的index找不到5会报错。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值