python pandas - 如何根据另一列的值来应用函数

Question

我有一个像图像一样的DataFrame：

我需要在DataFrame中添加另一个列来计算＆＃34; gram＆＃34;对于每种产品，根据不同的数量单位。 那么，怎么办？

Answer 1

IIUC如果列UNIT仅lb且oz使用numpy.where：

df['gram'] = np.where(df.UNIT == 'lb', df.number / 0.0022046, df.number / 0.035274)

公式：

ounces-to-grams
pounds-to-grams