python - 使用 dict 查找向量化 for 循环

Question

我有以下 pandas 数据框，我想通过将数据列的每一行乘以一个适当的因子来将单位标准化为 mg/kg:

df
Out[28]: 
  analyte  data   unit
0       A  2.34  mg/kg
1       B  5.89  ng/kg
2       C  1.25  ug/kg
3       D  8.65  mg/kg

我将如何对以下代码进行向量化(因为实际数据框要大一些):

units = {
        "mg/kg": 1,
        "ug/kg": 1e+3,
        "ng/kg": 1e+6
        }

for index, row in df.iterrows():
    df.at[index,'data'] = row['data'] * units[row['unit']]

df['unit'] = 'mg/kg'

Answer 1

那么你可以使用 map和 mul :

df['data'] = df['unit'].map(units).mul(df['data'])