从存在值中查找索引(检查是否存在)

时间:2019-05-27 11:27:32

标签: python python-3.x numpy indexing

我正在使用python-3.x,我想检查字典中是否存在该值如果存在,那么我想在字典中查找并打印该值的索引号

这是我的代码示例:

# import numpy
import numpy as np

# my first array
my_array_1 = np.array([[ 1  , 2  , 3  ],
                       [ 32 , 42 , 11 ],
                       [ 9  , 21 , 22 ],
                       [ 9  , 21 , 22 ],
                       [ 9  , 21 , 22 ],
                       [ 32 , 42 , 11 ],
                       [ 1  , 2  , 3  ]])


# here I want to find the unique values from my_array_1
indx = np.unique(my_array_1, return_index=True, return_counts= True,axis=0)


#save the result to dictionary
dic_t= {"my_array_uniq":indx[0],
       "counts":indx[1]}


# my 2nd array
my_array_2 = np.array([[ 1  , 2   , 3  ],
                       [ 32 , 422 , 11 ],
                       [ 9  , 221 , 22 ],
                       [ 9  , 221 , 22 ],
                       [ 9  , 21  , 22 ],
                       [ 32 , 242 , 11 ],
                       [ 1  , 22  , 3] ])



for i in range (len(my_array_2)):

# here I want to check if the vlue exist or not


    if any((my_array_2[i] == j).all() for j in dic_t["my_array_uniq"]):

######(if it exists print the index number of the existing vlue in dic_t["my_array_uniq"])######

        print (50*"*", i, "Yes")
        print (my_array_2[i], "\n")
        print (dic_t["my_array_uniq"])

    else:
        print (50*"*", i, "No")        
        dic_t["my_array_uniq"] = np.vstack((dic_t["my_array_uniq"], my_array_2[i])) 

我所需要的只是我尽力找到正确方法的索引号,但是它们都不起作用...

1 个答案:

答案 0 :(得分:0)

问题在于,在测试元素是否存在的行中,您没有将索引保留在找到它的位置。

您应该使用枚举来处理索引和列表中的值:

for i, a in enumerate(my_array_2):

# here I want to check if the vlue exist or not

    ix = [k for k,j in enumerate(dic_t["my_array_uniq"]) if (a == j).all()]    
    if ix:

######(if it exists print the index number of the existing vlue in dic_t["my_array_uniq"])######

        print (50*"*", i, "Yes", "at", ix[0])
        print (my_array_2[i], "\n")
        print (dic_t["my_array_uniq"])

    else:
        print (50*"*", i, "No")        
        dic_t["my_array_uniq"] = np.vstack((dic_t["my_array_uniq"], my_array_2[i]))