请教如何根据条件使用 numpy 函数生成一个新的数组

import pandas as pd
import numpy as np

A = np.array([3.33, 1.18, 5.42, 4.12])
B = np.array([449, 3676, 10975, 10975])
df = pd.DataFrame(data={"A": A, "B": B})
df["C"] = df.groupby("B")["A"].transform("max")
df["C"] = df.apply(lambda row: 0 if row["C"] > row["A"] else row["C"], axis=1)
df

随手写的，应该还有更简洁的写法

@TimePPT 谢谢帮助，我觉得还是写个函数来解决这个问题。