简体   繁体   中英

Extract lists from a list in dictionary

Suppose I have this dictionary:

self.dict = {'A':[[10, 20],[23,76,76],[23,655,54]], 'B':[30, 40, 50], 'C':[60, 100]}

Where the key 'A' is a list of lists. I want to get only the first 2 lists of 'A', ie [10, 20],[23,76,76] . I tried the idea of looping but it does not work well. :

class T(object):
    def __init__(self):
            self.dict = {'A':[[10, 20],[23,76,76],[23,655,54]], 'B':[30, 40, 50], 'C':[60, 100]}

        def output(self):
            for i in self.dict:
                for j in self.dict[i]:
                        first_two_lists = j
                    print ("%s" % (first_two_lists))

if __name__ == '__main__':
        T().output()

How can I get that ?

>>> d = {'A':[[10, 20],[23,76,76],[23,655,54]], 'B':[30, 40, 50], 'C':[60, 100]}
>>> d['A'][:2]
[[10, 20], [23, 76, 76]]

Using list slicing:

>>> d = {'A':[[10, 20],[23,76,76],[23,655,54]], 'B':[30, 40, 50], 'C':[60, 100]}
>>> d.get('A')[:2]
[[10, 20], [23, 76, 76]]

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM