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.