Методы добавления одного столбца в массивы NumPy

Чтобы добавить один столбец в NumPy, вы можете использовать различные методы. Вот несколько часто используемых подходов:

Метод 1: использование numpy.concatenate()

import numpy as np
# Create a 2D array
arr = np.array([[1, 2], [3, 4], [5, 6]])
# Create a new column
new_col = np.array([7, 8, 9])
# Add the new column using numpy.concatenate()
new_arr = np.concatenate((arr, new_col[:, np.newaxis]), axis=1)
print(new_arr)

Метод 2: использование numpy.hstack()

import numpy as np
# Create a 2D array
arr = np.array([[1, 2], [3, 4], [5, 6]])
# Create a new column
new_col = np.array([7, 8, 9])
# Add the new column using numpy.hstack()
new_arr = np.hstack((arr, new_col[:, np.newaxis]))
print(new_arr)

Метод 3: использование numpy.column_stack()

import numpy as np
# Create a 2D array
arr = np.array([[1, 2], [3, 4], [5, 6]])
# Create a new column
new_col = np.array([7, 8, 9])
# Add the new column using numpy.column_stack()
new_arr = np.column_stack((arr, new_col))
print(new_arr)

Эти методы позволяют добавить дополнительный столбец в массив NumPy. Не забудьте настроить форму нового столбца с помощью np.newaxisили изменить его форму, чтобы она соответствовала размерам существующего массива.