在嵌套字典值中切出第N个元素

时间:2016-11-30 10:51:54

标签: python arrays pandas numpy

祝大家明天第一个冬天好。虽然,为了我的好日子,我需要让我的嵌套字典正确...

DATA。 (字典,其中键是元组,值是2维numpy数组)

lst = [(0), (1), (1,2), (1,2,3), (1,2,3,4)]
array = np.random.random((5,2))
dictionary = dict(zip(lst, array))

问题。如何删除字典值中的第一个元素(数组的第一个维度)?或者我如何切割字典,以便只保留值中的第二个元素(数组的第二个维度)?

1 个答案:

答案 0 :(得分:1)

如果我理解正确,这可以帮到你的工作:

lst = [(0), (1), (1,2), (1,2,3), (1,2,3,4)]
array = np.random.random((5,2))
dictionary = dict(zip(lst, array))
element = 0
dictionary = dict(zip(dictionary.keys(), map(lambda x: np.delete(x, element), dictionary.values())))