关于np.where详解

动机是学习机器学习的时候遇到了困惑:

Ymean = np.zeros((movies, 1))
Ynorm = np.zeros((movies, users))

for i in range(movies):
    idx = np.where(R[i,:] == 1)[0]#这里的[0]表示返回的是行索引,还是列索引?
    Ymean[i] = Y[i,idx].mean()
    Ynorm[i,idx] = Y[i,idx] - Ymean[i]

Ynorm.mean(), idx

网上看了许多资料,说np.where(...)[0]表示返回的是行索引;np.where(...)[1]表示返回的是列索引。所以我就产生了困惑,这个解释在我这里好像行不通。
我就自己实验了一下,终于明白了其中奥义。截图如下,无需多言。
在这里插入图片描述
在这里插入图片描述
我是用jupyter notebook做的验证,文件放在链接里了,有需要自己跑一下,非常简单。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值