-
Notifications
You must be signed in to change notification settings - Fork 0
Open
Description
ImportError: IProgress not found. Please update jupyter and ipywidgets
from tqdm import tqdm
import pandas as pd
import numpy as np
from numba import jit
import numba
import time
import matplotlib.pyplot as plt
from numba_progress import ProgressBar
#make empty column
subscriber['all_user_id'] = 0
#change dtype to int
subscriber['user_id'] = subscriber['user_id'].astype(int)
customer_data['id_billing_user'] = customer_data['id_billing_user'].astype(int)
customer_data['__newID'] = customer_data['__newID'].astype(int)
subscriber['all_user_id'] = subscriber['all_user_id'].astype(int)
#convert pandas columns to numpy
subscriber_user_id = subscriber['user_id'].to_numpy()
billing_user_id = customer_data['id_billing_user'].to_numpy()
newID = customer_data['__newID'].to_numpy()
subscriber_all_user_id = subscriber['all_user_id'].to_numpy()
num_iterations = subscriber_user_id.shape[0]
@numba.jit
def do_it_numba(subscriber_user_id, billing_user_id, newID, subscriber_all_user_id, progress_proxy):
for i in range(subscriber_user_id.shape[0]):
for j in range(subscriber_user_id.shape[0]):
if subscriber_user_id[i] == billing_user_id[j]:
subscriber_all_user_id[j] = j
#print(j)
else:
pass
progress_proxy.update(1)
return subscriber_all_user_id
with ProgressBar(total=num_iterations) as progress:
numba_function(num_iterations, progress)
subscriber_all_user_id = do_it_numba(subscriber_user_id, billing_user_id, newID, subscriber_all_user_id, progress_proxy)
Metadata
Metadata
Assignees
Labels
No labels