Remove and add values to numpy array

Question

Is there a more efficient way to remove the 0 from the beginning and insert the 20 at the end and retain the shape (1, 20)?

# What I have.
array = np.arange(20)[np.newaxis]
print(array.shape, array)

# Remove 0 from the beginning and add 20 to the end.
array = np.append(array[0, 1:], np.array([[20]]))
print(array)
array = array[np.newaxis]
print(array.shape, array)

Output:

(1, 20) [[ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19]]
[ 1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20]
(1, 20) [[ 1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20]]

A DUBEY · Accepted Answer · 2021-08-04 14:54:58Z

3

You can just select a subset of the current array excluding the first element and then add 20 or whatever scalar you want at the end.

x = np.append(array[:,1:],[[20]], axis=1)

answered Aug 4, 2021 at 14:54

A DUBEY

8666 silver badges20 bronze badges

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

Comments

23vil · Accepted Answer · 2021-08-04 15:30:11Z

1

Maybe like this:

 array= np.linspace(start=1, stop=20, num=20, endpoint=True, dtype=int)[np.newaxis]
 print(array.shape, array)

answered Aug 4, 2021 at 15:30

23vil

1311 silver badge10 bronze badges

Collectives™ on Stack Overflow

Remove and add values to numpy array

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related