How to make a list of integers that is the sum of all the integers from a set of lists in a dict?

前端 未结 2 387
渐次进展
渐次进展 2021-01-12 17:30

Let\'s assume I have a created a dict that is made up of n keys. Each key is mapped to a list of integers of a consistent length. What I want to make now is a new list that

2条回答
  •  北海茫月
    2021-01-12 18:01

    Depending on your use-case you might want to consider using an adequate library for more general/complex functionality.

    numpy: general scientific computing

    import numpy as np
    
    my_dict = {'a': [1, 2, 3, 4], 'b': [2, 3, 4, 5], 'c': [3, 4, 5, 6]}
    
    arr = np.array(list(d.values()))
    # [[1 2 3 4]
    #  [2 3 4 5]
    #  [3 4 5 6]]
    
    arr.sum(axis=0)
    # [ 6  9 12 15]
    

    pandas: data-analysis toolkit

    import pandas as pd
    
    my_dict = {'a': [1, 2, 3, 4], 'b': [2, 3, 4, 5], 'c': [3, 4, 5, 6]}
    
    df = pd.DataFrame(my_dict)
    #    a  b  c
    # 0  1  2  3
    # 1  2  3  4
    # 2  3  4  5
    # 3  4  5  6
    
    df.sum(axis=1)
    # 0     6
    # 1     9
    # 2    12
    # 3    15
    

提交回复
热议问题