Вернуть индекс чисел больше k для каждого из подсписок в списке в Python

Question

Ввод: найти индекс чисел больше 50 в подсписках

a = [[10,40,90],[120,30,200],[70,90,100]]

Желаемый результат:

index_of_values_greater_than_50 = [[2][0,2][0,1,2]]

Answer 1

output_list = []

a = [[10,40,90],[120,30,200],[70,90,100]]

for array in a:
   counter = 0
   new_list = []
   while counter < len(array):
      if array[counter] > 50:
         new_list.append(counter)
      counter += 1
   output_list.append(new_list)

print(output_list)

ВЫХОД (по желанию):

[[2], [0, 2], [0, 1, 2]]   


     
     

Answer 2

index_of_values_greater_than_50 = []
for x in a:
 temp = [i for i, y in enumerate(x) if (y>50)
 index_of_values_greater_than_50.append(temp)