python - 应用一个函数来翻译熊猫数据框中的一列，条件是其他列

Question

data 是一个 pandas 数据框，其中 language 和 config['TEXT FIELD'] 是列。我想将文本栏中的某些评论翻译成英文，我正在使用函数 dfApply

import goslate
def dfApply(row):
    if row["langauge"] == 'en':
       return row[config['TEXT FIELD']]
    else:
       return gs.translate(row[config['TEXT FIELD']], 'en')


gs = goslate.Goslate()
data['english_text'] = data.apply(dfApply, axis=1)

但是编译器显示了以下错误

KeyError: ('langauge', 'occurred at index 0')

Answer 1

这样的事情可能是一种更简单的方法。

not_en = data["language"] != "en"
trans = translate(data[config['row']], "en")
col = config['row']
data.loc[not_en, col] = trans[not_en]