數據框內的熊貓數據框

Question

我正在嘗試訓練多對象分類器。 為此，我將數據集信息存儲到了當前的熊貓數據框架中，如下所示：

|IMAGE_PATHS---|LABELS------------------------------------------|
|path_to_image1|[[c11 x11 y11 w11 h11],[c12 x12 y12 w12 h12]...]| 
|path_to_image2|[[c21 x21 y21 w21 h21],[c22 x22 y22 w22 h22]...]| 
|...

但是以這種方式擁有它並不容易。 例如，如果我想查看圖像中標記的所有獨角獸，則需要遍歷每行中的所有元素並尋找它們。 如果此標簽為DataFrames，則可以輕松將其過濾為df[df["label"] == "unicorn"] 。

那么，有沒有一種方法可以輕松地在此DataFrame中創建一個DataFrame或其他有趣的技巧？

Answer 1

如果您的標簽只是嵌套列表，則可以執行以下操作：

df[df['LABELS'].apply(lambda x: 'unicorn' in [item for sublist in x for item in sublist])]

這會將子列表拉平到lambda函數中的單個列表中，然后檢查其是否包含“獨角獸”，對df進行遮罩，最后返回過濾的df