supplement numpy array in python 3

Question

How do I supplemented by a numpy array of zeros?

arr = np.array([1,10])

if len(arr) < size:
  # supplemented array of zeros to the size

For example size = 5;

if array = [1,2,3].
output array = [1,2,3,0,0]

Should I call fill or make a new array like arr + [0]*(size-len)?

Which is faster?

mgilson · Accepted Answer · 2016-11-28 04:58:16Z

1

I'd use ndarray.resize:

>>> a = np.array([1,2,3])
>>> a.resize((5,))
>>> a
array([1, 2, 3, 0, 0])

edited Nov 28, 2016 at 4:58

answered Nov 28, 2016 at 4:56

mgilson

312k70 gold badges656 silver badges722 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Julien Over a year ago

or use numpy.pad... docs.scipy.org/doc/numpy/reference/generated/numpy.pad.html

mgilson Over a year ago

@JulienBernu -- Yeah, I just looked for a np.resize -- It turns out that there is np.resize and also ndarray.resize. The former repeats the array which isn't what OP wants, but the latter pads with 0 as OP wants, so I don't see any good reason not to use it... :-)

Collectives™ on Stack Overflow

supplement numpy array in python 3

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related