Chame return_weights() para retornar as palavras com maior peso para esse documento.
Chame set() no filter_list retornado para remover números duplicados.
Chame words_to_filter, passando os seguintes parâmetros: vocab para o parâmetro vocab, tfidf_vec.vocabulary_ para o parâmetro original_vocab, text_tfidf para o parâmetro vector e 3 para capturar as 3 palavras com maior top_n peso de cada documento.
Por fim, passe esse conjunto filtered_words para uma lista para usar como filtro do vetor de texto.

Selecionando features usando vetores de texto