def get_exist_list(): exist_tup = cursor.fetchall() exists = list() for a_item in exist_tup: exists.append(a_item[0]) return exists
cursor.execute( '''UPDATE UpdateFilm SET linear_predict=%s,linear_test=%s, lasso_predict=%s,lasso_test=%s,knn_predict=%s,knn_test=%s,poly_predict=%s,poly_test=%s WHERE imdb_filmID = %s''', (scoredict['linear_predict'], scoredict['linear_test'], scoredict['lasso_predict'], scoredict['lasso_test'], scoredict['knn_predict'], scoredict['knn_test'], scoredict['poly_predict'], scoredict['poly_test'], filmid)) db.commit() cursor.execute('SELECT click_times,gross ' 'FROM FilmDB,TrailerClick ' 'WHERE FilmDB.imdb_filmID=TrailerClick.imdb_filmID ' 'AND gross>4*TrailerClick.click_times') clicks = cursor.fetchall() X_R1 = list() y_R1 = list() for click_time, gross in clicks: X_R1.append(click_time) y_R1.append(gross) X_data = np.array(X_R1).reshape(-1, 1) y_data = np.array(y_R1).reshape(-1, 1) X_train, X_test, y_train, y_test = train_test_split(X_data, y_data, test_size=0.2, random_state=4) cursor.execute( 'SELECT TrailerClick.imdb_filmID,max(click_times) FROM UpdateFilm,TrailerClick WHERE TrailerClick.imdb_filmID=UpdateFilm.imdb_filmID GROUP BY UpdateFilm.imdb_filmID' ) parameters = cursor.fetchall()