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)
获取数列ndarray中指定元素的索引
最新推荐文章于 2022-11-14 22:15:19 发布