使用 .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 中，& 等位运算符的优先级高于== 等比较运算符。