How to add an extra column to a NumPy array

by

np.r_[ ... ] and np.c_[ ... ]
are useful alternatives to vstack and hstack,
with square brackets [] instead of round ().
A couple of examples:

: import numpy as np
: N = 3
: A = np.eye(N)

: np.c_[ A, np.ones(N) ]              # add a column
array([[ 1.,  0.,  0.,  1.],
       [ 0.,  1.,  0.,  1.],
       [ 0.,  0.,  1.,  1.]])

: np.c_[ np.ones(N), A, np.ones(N) ]  # or two
array([[ 1.,  1.,  0.,  0.,  1.],
       [ 1.,  0.,  1.,  0.,  1.],
       [ 1.,  0.,  0.,  1.,  1.]])

: np.r_[ A, [A[1]] ]              # add a row
array([[ 1.,  0.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  0.,  1.],
       [ 0.,  1.,  0.]])
: # not np.r_[ A, A[1] ]

: np.r_[ A[0], 1, 2, 3, A[1] ]    # mix vecs and scalars
  array([ 1.,  0.,  0.,  1.,  2.,  3.,  0.,  1.,  0.])

: np.r_[ A[0], [1, 2, 3], A[1] ]  # lists
  array([ 1.,  0.,  0.,  1.,  2.,  3.,  0.,  1.,  0.])

: np.r_[ A[0], (1, 2, 3), A[1] ]  # tuples
  array([ 1.,  0.,  0.,  1.,  2.,  3.,  0.,  1.,  0.])

: np.r_[ A[0], 1:4, A[1] ]        # same, 1:4 == arange(1,4) == 1,2,3
  array([ 1.,  0.,  0.,  1.,  2.,  3.,  0.,  1.,  0.])

(The reason for square brackets [] instead of round ()
is that Python expands e.g. 1:4 in square —
the wonders of overloading.)

More Related Contents:

  1. Convert a 3D array to RGB valued array [closed]
  2. Unable to allocate array with shape and data type
  3. What is the difference between NaN and None?
  4. Performance of Pandas apply vs np.vectorize to create new column from existing columns
  5. NumPy selecting specific column index per row by using a list of indexes
  6. Coalesce values from 2 columns into a single column in a pandas dataframe
  7. scipy: savefig without frames, axes, only content
  8. Interpolate NaN values in a numpy array
  9. Numpy ‘smart’ symmetric matrix
  10. Fitting a 2D Gaussian function using scipy.optimize.curve_fit – ValueError and minpack.error
  11. No module named when using PyInstaller
  12. ‘DataFrame’ object has no attribute ‘sort’
  13. Immutable numpy array?
  14. N-D version of itertools.combinations in numpy
  15. figure of imshow() is too small
  16. Binary random array with a specific proportion of ones?
  17. python recursive vectorization with timeseries
  18. Difference between np.random.seed() and np.random.RandomState()
  19. Loading UTF-8 file in Python 3 using numpy.genfromtxt
  20. What does a colon and comma stand in a python list?
  21. convert string representation of array to numpy array in python
  22. How to share numpy random state of a parent process with child processes?
  23. Flattening a list of NumPy arrays?
  24. Fastest pairwise distance metric in python
  25. How do I get Pylint to recognize NumPy members?
  26. Is there an analysis speed or memory usage advantage to using HDF5 for large array storage (instead of flat binary files)?
  27. python tilde unary operator as negation numpy bool array
  28. Sieve of Eratosthenes – Primes between X and N
  29. numpy arange: how to make “precise” array of floats?
  30. Sort array’s rows by another array in Python

Leave a Comment