Pandas 第一个不返回 KeyError 的值

Question

我有一个数据框，其中有列

A

或

B

我希望有一种使用

or

运算符的语法来具有如下所示的 stntax：

Value = df.at['Total','A'] or df.at['Total','B']

但我仍然收到

KeyError

异常。有没有一种简便的方法来实现这一点，而不是做类似的事情：

if 'A' in df.columns:
    Value = df.at['Total','A']
else:
    Value = df.at['Total','B']

Answer 1

如果您的 pandas 数据框始终具有相同顺序的相同数量的列，您可以使用列索引而不是名称。例如，要抓取整个第一列：

df[[1]]

或者获取第一行中的列条目：

df.iloc[0, 1]