Create dictionary from results of DataFrame in pandas

前端 未结 5 1443
旧时难觅i
旧时难觅i 2021-01-16 16:35

I have a dataframe with results as below. Sample dataframe shown actual one is much larger. I want to get a dictionary (or another structure if it will be faster) with the

5条回答
  •  再見小時候
    2021-01-16 17:09

    You can use .apply

    df.apply(lambda x: list(x.dropna().index), axis=1).to_dict()       #Updated answer
    # Or dict(df.apply(lambda x: list(x.index[~x.isnull()]), axis=1))  #Original answer
    

    Output:

    {1: ['MSFT'], 2: ['GOOG', 'AMZN'], 3: ['AAPL', 'AMZN', 'FB'], 4: ['FB']}
    

提交回复
热议问题