Numpy: Delete items from array

20 August 2020

Code

# Import library
import numpy as np

# Create an array
x = np.array([[1,2,3],
             [3,999,4],
             [5,6,7]]
            )
print(x)
[[  1   2   3]
 [  3 999   4]
 [  5   6   7]]

.

Delete using one integer:

# Using one integer
y0 = np.delete(x, 1, axis=0) # integer --> 1
y1 = np.delete(x, 1, axis=1) # integer --> 1

# Output
print('Delete (axis=0):\n',y0)
print('Shape:', y0.shape, '\n')

print('Delete (axis=1):\n',y1)
print('Shape:', y1.shape)
Delete (axis=0):
 [[1 2 3]
 [5 6 7]]
Shape: (2, 3) 

Delete (axis=1):
 [[1 3]
 [3 4]
 [5 7]]
Shape: (3, 2)

.

Delete using 1-D array of sorted integers:

# Using 1-D array of sorted integer
y0 = np.delete(x, [0,2], axis=0) # array --> [0,2]
y1 = np.delete(x, [0,2], axis=1) # array --> [0,2]

# Output
print('Delete (axis=0):\n',y0)
print('Shape:', y0.shape, '\n')

print('Delete (axis=1):\n',y1)
print('Shape:', y1.shape)
Delete (axis=0):
 [[  3 999   4]]
Shape: (1, 3) 

Delete (axis=1):
 [[  2]
 [999]
 [  6]]
Shape: (3, 1)






Any errors in code above?
Please send a message.