Just style, there is no need for the np.where here, the indexing will do that already internally.