np.where(df["A"]>45, ## condition "Class A", ## if condition is True "Class B" ## if condition is False ) ## Run-time: 2.1 s