替换numpy数组

Question

我有一个数组a和一个数组b，看起来像这样：

a = np.array([[ 1.,  0.,  0.,  0.,  0.],
              [ 1.,  0.,  0.,  0.,  0.],
              [ 3.,  0.,  0.,  0.,  0.]])
b = np.array([[2, 3, 4, 5],[2,3,4,5]])

现在我想在行中替换第二个a到第五个值，其中第一个元素等于1.所需的输出：

a = np.array([[ 1.,  2.,  3.,  4.,  5.],
              [ 1.,  2.,  3.,  4.,  5.],
              [ 3.,  0.,  0.,  0.,  0.]])

我试过了a[a[:,0] == 1][:,1:] = b，但它仍然给了我

a = np.array([[ 1.,  0.,  0.,  0.,  0.],
              [ 1.,  0.,  0.,  0.,  0.],
              [ 3.,  0.,  0.,  0.,  0.]])

我做错了什么？