Se da el siguiente número:
path = np.array([['S','A','N','V','T'],
['S','R','Z','V','W'],
['S','D','C','E','Y'],
['S','W','C','E','Y'],
['S','Q','R','E','B'],
['S','Q','R','Z','Z']])
Es necesario comparar los valores en la columna 3 y devolver listas de índices de filas donde existe la igualdad. En el ejemplo anterior en la columna 3 existen dos ecuaciones (valores V
y E
) por lo tanto, el resultado devuelto debe ser:
[[0,1],[2,3,4]]
si no sabe cuáles son los valores repetidos
value_dict = {}
for i in range(path.shape[0]):
if(path[i,3]) in value_dict :
value_dict [path[i,3]].append(i)
else:
value_dict [path[i,3]]=[i]
index_list = []
for k in value_dict:
if len(value_dict[k])>1:
index_list.append(value_dict[k])
la salida es [[0, 1], [2, 3, 4]]
Este artículo se recopila de Internet, indique la fuente cuando se vuelva a imprimir.
En caso de infracción, por favor [email protected] Eliminar
Déjame decir algunas palabras