將矢量值分配給numpy中的段

Question

我試圖使用numpy（和numpy）使用cutoffs中的邊來分割矢量y。將y和g定義爲列向量，但將a作爲行向量返回。運行g += a.transpose()通過引入100x100陣列生成形狀一致性錯誤（複製如下）。必須有一個更加優雅的方式來完成這個任務。謝謝。

y = np.random.uniform(0,1,100) 
cutoffs = np.random.uniform(0,1,3) 
cutoffs.sort() 
g = np.zeros(y.size) 
for c in np.hstack([ cutoffs , 1. ]): 
    a = np.array([ y < c ]) 
    g += a.transpose() 


Traceback (most recent call last): 
    File "<stdin>", line 3, in <module> 
ValueError: non-broadcastable output operand with shape (100,) doesn't match the broadcast shape (100,100) 

Answer 1

使用

a = (y < c).reshape(-1) 
g += a