获取数列ndarray中指定元素的索引

import numpy as np
from sklearn import datasets
from sklearn.cluster import KMeans

A = np.array([1,2,2,1,1,3,3])

a = np.where(A==1)[0]
b = np.where(A==2)[0]
C = np.hstack((a,b))
print(C)