df.A.apply(assign_class) ## Run-time: 12.5 s
Hosted onDeepnote