Why does an assignment for double-sliced numpy arrays not work?

前端 未结 4 2495
甜味超标
甜味超标 2021-02-20 05:18

why do the following lines not work as I expect?

import numpy as np
a = np.array([0,1,2,1,1])
a[a==1][1:] = 3
print a
>>> [0 1 2 1 1]
# I would expect [         


        
4条回答
  •  暖寄归人
    2021-02-20 05:49

    It appears you simply can't do an assignment through a double-slice like that.

    This works though:

    a[numpy.where(a==1)[0][1:]] = 3
    

提交回复
热议问题