python - 在对象调用中使用变量

Question

我正在尝试在扩展方法调用中使用变量。我想用变量替换对象引用。

import pandas
data = pandas.read_csv('sample1.csv')
zones = list(data.columns.values)
for zone in zones:
    print(zone)
    varzone = data.zone.tolist()
    print(varzone)

假设区域等于['name', 'surname']

我希望它运行为

varzone = data.name.tolist()

而不是:

varzone = data.zone.tolist()

错误:

Traceback (most recent call last):
  File "script1.py", line 10, in <module>
    varzone = data.zone.tolist()
  File "/usr/lib64/python2.7/site-packages/pandas/core/generic.py", line 4376, in __getattr__
    return object.__getattribute__(self, name)
AttributeError: 'DataFrame' object has no attribute 'zone'

Answer 1

Python 正在查找名为 zone 的属性，而不是按存储在 zone 中的名称查找列。正确的做法是这样的:

for zone in zones:
    print(zone)
    varzone = data[zone].tolist()
    print(varzone)