如何删除列表中的那些列元素的行？

Question

在上图中，最后一列（“位置”）是一个列表。对于以下情况，我需要删除这些行：

if first item in the list 'location' is greater than 60.0, those rows are not needed for me

如果我使用：

for i in range(len(output)):
    trip_df = output.drop(output[ output['location'][i][0] > 60].index)

错误是：KeywordError: True

我不能简单地放置 output['location'][0]，因为它只表示第一行，我如何在有或没有 for 循环的情况下做到这一点？

P.S : 对不起图片，我不知道如何将值放在表格中

Answer 1

使用str方法获取第一项：

df = pd.DataFrame({"Location":[[61,50],[117,24],[3,5]],     #dummy data
                   "Values":[2,4,6]})

print (df[~df["Location"].str[0].gt(60)])

  Location  Values
2   [3, 5]       6