Write a NumPy program to find unique rows in a NumPy array
import numpy as np x = np.array([[20, 20, 20, 0], [0, 20, 20, 20], [0, 20, 20, 20], [20, 20, 20, 0], [10, 20, 20,20]]) print("Original array:") print(x) y = np.ascontiguousarray(x).view(np.dtype((np.void, x.dtype.itemsize * x.shape[1]))) _, idx = np.unique(y, return_index=True) unique_result = x[idx] print("Unique rows of the above array:") print(unique_result)
Sample Output:
Original array: [[20 20 20 0] [ 0 20 20 20] [ 0 20 20 20] [20 20 20 0] [10 20 20 20]] Unique rows of the above array: [[ 0 20 20 20] [10 20 20 20] [20 20 20 0]]
total answers (1)
start bookmarking useful questions and collections and save it into your own study-lists, login now to start creating your own collections.
Sample Output:
need an explanation for this answer? contact us directly to get an explanation for this answer