根据其中一列的值从DataFrame中选择行

时间:2018-04-15 12:54:51

标签: python pandas dataframe where keyerror

DataFrame named "result"

**我正在尝试从您在附加图片中看到的DataFrame(名为“values == 0”获取result行。 像这样的SQL查询将是这样的: SELECT * FROM RESULT WHERE column_name等于some_value 我的DataFrame有2列。第一个是我认为名为'Index',第二个名为'0'(名称是默认的 - 没有打算改变)

因为第二列是二进制:根据餐厅评论(第1列)是正面还是负面(1或0)。我想只得到行(根据负面评论的两列)。 当我尝试代码时: df.loc[df['column_name'].isin(some_values)]或我的情况 result.loc[result['0'].isin(0)]

我收到错误:KeyError: '0'

有人可以帮忙吗? **

1 个答案:

答案 0 :(得分:1)

如果要选择列的值为' 0'是0,它应该简单:

result[result[0] == 0]

但是,正如sacul指出的那样,您应该检查列的名称是否为' str'或者' int'。