在obj中使用变量

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'，'姓氏']

我希望它能像以前一样运行

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)