python - 使用 .loc 在 Pandas 中进行多选

Question

我已成功使用 .loc 来分配已过滤数据集的列，但现在我需要使用两个过滤器来过滤数据集并收到以下错误。我正在尝试使用 & 运算符，就像您只是过滤一样。

TypeError: cannot compare a dtyped [object] array with a scalar of type [bool]

代码:

import pandas as pd
df = pd.DataFrame(data=[['Arlington', 'MA'],
                           ['Arlington', 'TX'],
                           ['Dallas', 'TX']],
                     columns=['City', 'State'])

df.loc[(df['State'] == 'TX' & df['City'] == 'Arlington'), 'New Column'] = 10

print df

期望的输出:

            City State  New Column
0      Arlington    MA         NAN
1      Arlington    TX          10
2         Dallas    TX         NaN

Answer 1

您需要将比较结果括起来，如 (df['State'] == 'TX') & (df['City'] == 'Arlington')。在 Python 中，像 & 这样的按位运算符比 == 这样的比较运算符具有更高的优先级。