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