Genere un método el cual acepte como argumento una matriz de cadenas y genere otra matriz del mismo orden en donde almacene en cada posición la cantidad de vocales que contiene cada palabra en la misma posición

def count_vowels(words):
  vowels = {'a', 'e', 'i', 'o', 'u'}
  new_words = []
  for word in words:
    count = 0
    for letter in word:
      if letter in vowels:
        count += 1
    new_words.append(count)
  return new_words
    
count_vowels(['hello', 'world'])