Не удалось добавить элемент в массив numpy

Question

Я пытаюсь добавить значения, сгенерированные функцией calc_class, но это не работает, и я не знаю причины. Я попытался использовать numpy.append, numpy.insert и встроенную функцию Python append безуспешно.

Это мой кусок кода:

def calc_class(test): 
    expec = [] 
    for new in test: 
     prob_vector = np.zeros((len(voc)), dtype=bool) #define a 'True' array to store class probabilities 
     words_in_new = new[0].split() #split the new email into words 
     words_in_new = list(set(words_in_new)) #remove duplicated words 
     i = 0 
     for voc_word in voc: #for each element in voc 
      if voc_word in words_in_new: 
       prob_vector[i] = True #set the ith element of prob_vector to True, if voc element is in word 
      else: 
       prob_vector[i] = False #set the ith element of prob_vector to False, otherwise 
      i += 1 
     prob_ham = 1 
     for i in range(len(prob_vector)): 
      if prob_vector[i] == True: 
       prob_ham *= ham_class_prob[i] 
      else: 
       prob_ham *= (1 - ham_class_prob[i]) 
     # alternative:  np.prod(ham_class_prob[np.where(prob_vector==True)]) * np.prod(1- ham_class_prob[np.where(prob_vector==False)]) 

     prob_spam = 1 
     for i in range(len(prob_vector)): 
      if prob_vector[i] == True: 
       prob_spam *= spam_class_prob[i] 
      else: 
       prob_spam *= (1 - spam_class_prob[i]) 

     p_spam = 0.3 
     p_ham = 1 - p_spam 

     p_spam_given_new = (prob_spam * p_spam)/(prob_spam * p_spam + prob_ham * p_ham) # Bayes theorem 
     print('p(spam|new_email)=', p_spam_given_new[0]) 
     expec.append(p_spam_given_new[0]) 
     print(expec) 

Проблема заключается в том, что print(expect) печатает пустой массив.

Answer 1

Вы можете использовать pdb do debug (или ipdb для ipython).

from pdb import set_trace 

использование "set_trace()" вместо "печати ('р (спам | NEW_EMAIL) =', p_spam_given_new [0])" (третья строка подсчитываются с конца строки), чем запустить свой код. Он остановится на этой строке, и вы можете запустить любой код python, например «print (p_spam_given_new)» или просто «p_spam_given_new», вы также можете проверить «prob_spam», «p_spam» или любую другую переменную, которую вы хотите проверить.