Как добавить их в матрицу?

Question

У меня есть массив:

X = [[2, 2, 2], 
    [3, 3, 3], 
    [4, 4, 4]] 

Мне нужно добавить дополнительный столбец в Numpy массив и заполнить его с тех использованием hstack и перекраивать. Как что:

X = [[2, 2, 2, 1], 
    [3, 3, 3, 1], 
    [4, 4, 4, 1]] 

Что я делаю:

X = np.hstack(X, np.ones(X.reshape(X, (2,3)))) 

И получите ошибку:

TypeError: only length-1 arrays can be converted to Python scalars 

Что проблема? Что я сделал не так?

Answer 1

Вот несколько способов с numpy.append, numpy.hstack или numpy.column_stack:

# numpy is imported as np 
>>> x 
array([[2, 2, 2], 
     [3, 3, 3], 
     [4, 4, 4]]) 
>>> np.append(x, np.ones([x.shape[0], 1], dtype=np.int32), axis=1) 
array([[2, 2, 2, 1], 
     [3, 3, 3, 1], 
     [4, 4, 4, 1]]) 
>>> np.hstack([x, np.ones([x.shape[0], 1], dtype=np.int32)]) 
array([[2, 2, 2, 1], 
     [3, 3, 3, 1], 
     [4, 4, 4, 1]]) 
>>> np.column_stack([x, np.ones([x.shape[0], 1], dtype=np.int32)]) 
array([[2, 2, 2, 1], 
     [3, 3, 3, 1], 
     [4, 4, 4, 1]]) 

Answer 2

Вы можете использовать numpy.insert():

>>> X 
array([[2, 2, 2], 
     [3, 3, 3], 
     [4, 4, 4]]) 

Ones в начале матрицы:

>>> X=np.insert(X,0,1.0,axis=1) 
>>> X 
array([[1, 2, 2, 2], 
     [1, 3, 3, 3], 
     [1, 4, 4, 4]]) 

Ones в конец матрицы

>>> X=np.insert(X,3,1.0,axis=1) 
>>> X 
array([[2, 2, 2, 1], 
     [3, 3, 3, 1], 
     [4, 4, 4, 1]])