C[..., (2)] = A[:, 3:][:, (None), :]
C[..., (3)] = np.arange(len(A) * 2).reshape(len(A), 2, 1) + 1
C = C.reshape(-1, 4)
C = C[C[:, (2)] != 0]
print(C)
Создайте новый массив numpy на основе условий, установленных в массиве numpy