Python & Pandas: How to query if a list-type column contains something?

前端 未结 5 863
攒了一身酷
攒了一身酷 2020-11-27 17:59

I have a dataframe, which contains info about movies. It has a column called genre, which contains a list of genres it belongs to. For example:

         


        
5条回答
  •  臣服心动
    2020-11-27 18:32

    A complete example:

    import pandas as pd
    
    data = pd.DataFrame([[['foo', 'bar']],
                        [['bar', 'baz']]], columns=['list_column'])
    print(data)
      list_column
    0  [foo, bar]
    1  [bar, baz]
    
    filtered_data = data.loc[
        lambda df: df.list_column.apply(
            lambda l: 'foo' in l
        )
    ]
    print(filtered_data)
      list_column
    0  [foo, bar]
    

提交回复
热议问题