Индекс соответствия для поиска другого значения в том же индексе - Python

Question

Я пытаюсь написать функцию, которая вычисляет число RI, ищет RI между 0 и 0,25, а затем сопоставляет их с высотами, на которых были вычислены эти числа RI.

Я знаю, что в sig_ri_final есть это 4 значения, которые мне теперь нужно найти, на которых они произошли.

Я пробовал:

    res = [test_list1.index(i) for i in test_list2]
    res = [key for key, val in enumerate(test_list1) 
                  if val in set(test_list2)] 


    def ri_level(T, tk, hi, P, Hu, ws, Wd):
g = 9.81
ri = (pot - pot[0]) * hi * g / ( pot * (u ** 2 + v ** 2) )
for value in ri:
    if value <= 0.25:
        significant_ri.append(value)
for sigval in significant_ri:
    if sigval >= 0:
        sig_ri_final.append(sigval)
for height in sig_ri_final:
    height_sigri.append(hi.index(sig_ri_final.index()))
return print(height__sigri)