How do I add rows and columns to a NUMPY array? How do I add rows and columns to a NUMPY array? numpy numpy

How do I add rows and columns to a NUMPY array?


This should do what you want (ie, using 3x3 array and 4x4 array to represent the two arrays in the OP)

>>> import numpy as NP>>> a = NP.random.randint(0, 10, 9).reshape(3, 3)>>> a>>> array([[1, 2, 2],           [7, 0, 7],           [0, 3, 0]])>>> b = NP.zeros((4, 4))

mapping a on to b:

>>> b[:3,:3] = a>>> b    array([[ 1.,  2.,  2.,  0.],           [ 7.,  0.,  7.,  0.],           [ 0.,  3.,  0.,  0.],           [ 0.,  0.,  0.,  0.]])


If you want zeroes in the added elements, my_array.resize((1600, 1000)) should work. Note that this differs from numpy.resize(my_array, (1600, 1000)), in which previous lines are duplicated, which is probably not what you want.

Otherwise (for instance if you want to avoid initializing elements to zero, which could be unnecessary), you can indeed use hstack and vstack to add an array containing the new elements; numpy.concatenate() (see pydoc numpy.concatenate) should work too (it is just more general, as far as I understand).

In either case, I would guess that a new memory block has to be allocated in order to extend the array, and that all these methods take about the same time.


No matter what, you'll be stuck reallocating a chunk of memory, so it doesn't really matter if you use arr.resize(), np.concatenate, hstack/vstack, etc. Note that if you're accumulating a lot of data sequentially, Python lists are usually more efficient.