读文件时如何添加进度条
import tqdm
def parse_all_seq(students):
all_sequences = []
for student_id in tqdm.tqdm(students, 'parse student sequence:\t'):
student_sequence = parse_student_seq(data[data.user_id == student_id])
all_sequences.extend([student_sequence])
return all_sequences
def parse_student_seq(student):
seq = student.sort_values('order_id')
s = [skills[q] for q in seq.skill_id.tolist()]
p = [problems[q] for q in seq.problem_id.tolist()]
a = seq.correct.tolist()
return s, p, a
# [(skill_seq_0, problem_seq_0, answer_seq_0), ..., (skill_seq_n, problem_seq_n, answer_seq_n)]
sequences = parse_all_seq(data.user_id.unique())
print(len(data.user_id.unique()))
呈现效果