在python中搜索索引位置

Question

cols = [2,4,6,8,10,12,14,16,18] # selected the columns i want to work with
df = pd.read_csv('mywork.csv')
df1 = df.iloc[:, cols]
b= np.array(df1)
b
outcome
b = [['WV5 6NY' 'RE4 9VU' 'BU4 N90' 'TU3 5RE' 'NE5 4F']
   ['SA8 7TA' 'BA31 0PO' 'DE3 2FP' 'LR98 4TS' 0]
   ['MN0 4NU' 'RF5 5FG' 'WA3 0MN' 'EA15 8RE' 'BE1 4RE']
   ['SB7 0ET' 'SA7 0SB' 'BT7 6NS' 'TA9 0LP' 'BA3 1OE']]

a = np.concatenate(b) #concatenated to get a single array, this worked well

a = np.array([x for x in a if x != 'nan'])
a = a[np.where(a != '0')] #removed the nan
print(np.sort(a)) # to sort alphabetically

#Sorted array
['BA3 1OE' 'BA31 0PO' 'BE1 4RE' 'BT7 6NS' 'BU4 N90'
'DE3 2FP' 'EA15 8RE' 'LR98 4TS' 'MN0 4NU', 'NE5 4F' 'RE4 9VU'
'RF5 5FG' 'SA7 0SB' 'SA8 7TA' 'SB7 0ET' 'TA9 0LP' 'TU3 5RE'
'WA3 0MN' 'WV5 6NY']

#Find the index position of all elements of b in a(sorted array)
def findall_index(b, a )
    result = []
    for i in range(len(a)):
       for j in range(len(a[i])):
           if b[i][j] == a:
               result.append((i, j))
    return result
print(findall_index(0,result))

我对python还是很陌生，我尝试在上面的a中找到b的所有元素的索引位置。下面的代码块似乎没有给我任何结果。请有人帮助我。

提前谢谢你。

Answer 1

解决此问题的一种方法是将 b 中的元素索引与实际元素压缩（创建对），然后仅根据元素对这个新数组进行排序。现在您有了从原始数组的索引到新排序数组的映射。然后，您可以遍历已排序的对以将当前索引映射到原始索引。

我强烈建议您自己编写代码，因为它会帮助您学习！