python - 使用列值从 Pandas DataFrame 获取数据

Question

>>> df = pd.DataFrame({'num_legs': [4, 2], 'num_wings': [0, 2]},
...                   index=['dog', 'hawk'])
>>> df
      num_legs  num_wings
dog          4          0
hawk         2          2
>>> for row in df.itertuples():
...     print(row)
...
Pandas(Index='dog', num_legs=4, num_wings=0)
Pandas(Index='hawk', num_legs=2, num_wings=2)

我正在使用 pandas.DataFrame.itertuples 解析 excel 工作表，这将在每次迭代时给我一个 pandas.DataFrame。考虑在每次迭代中返回的 pandas.DataFrame，如上所示。

现在关闭每个数据框 Pandas(Index='dog', num_legs=4, num_wings=0) 我想使用关键字 num_legs 访问这些值但是，在使用相同的方法后，我得到以下异常。

TypeError: tuple indices must be integers, not str

有人可以帮助了解如何直接使用列标题从数据框中检索数据。

Answer 1

我在使用变量时遇到了同样的错误。

v = 'num_legs'
for row in df.itertuples():
    print(row[v])

TypeError: tuple indices must be integers or slices, not str

使用df.itertuples() 并使用属性名称作为变量。

v = 'num_legs'
for row in df.itertuples():
    print(getattr(row, v))

最后 df.itertuples() 比 df.iterrows() 快。