Add a column

Write a NumPy program to add an extra column to an array.

[[4, 8, 0], [2, 0, 6]], column [5, 7]

[[4 8 0 5]
 [2 0 6 7]]

Approach

1. np.column_stack or np.hstack with a reshaped column.

Solution

import numpy as npa = np.array([[4, 8, 0], [2, 0, 6]])print(np.column_stack((a, [5, 7])))

[[4 8 0 5]
 [2 0 6 7]]

More numpy programs

← All 100 NumPy Programs