Python：如何将字符串作为函数的参数传递

Question

我有一个包含美国州缩写的数据框：

state
tx
ca
fl
wa
ny
az

我试图使用us.states模块获取全名

df['state_ru'] = df.state.apply(lambda x: us.states.x.upper().name)

它返回

AttributeError: module 'us.states' has no attribute 'x'

我不清楚如何将x传递到us.states模块以生成我正在寻找的内容。

预期结果：

state  state_ru
tx     Texas
ca     California
fl     Florida
wa     Washington
ny     New York
az     Arizona

Answer 1

us.states模块有一个可以使用的查找方法：

import us
df['state_ru'] = df.state.apply(lambda x: us.states.lookup(x).name)
df

Answer 2

是的，你可以使用getattr这样的

来做到这一点

getattr(obj, 'foo')

这相当于执行：obj.foo，当属性名称可变时非常有用。