"""
Understang heapq
"""
nums = [1, 8, 2, 23, 7, -4, 18, 23, 42, 37, 2]
# How to find max and min in the list??
In [2]: max(nums)
Out[2]: 42
In [3]: min(nums)
Out[3]: -4
#How to find n maximum numbers?
#this is not a well written .I'm lazy so is there a pythonic way?
def n_max(n,nums):
max_list=[]
if len(nums) == 0:
return (0)
elif len(nums) == 1:
return nums
elif len(nums) >1:
for i in range(n):
max_list.append(max(nums))
nums.remove(max(nums))
return max_list,len(max_list)
print n_max(11,nums)
#Using heapq
#The heapq module has two functions— nlargest() and nsmallest() —that do exactly and more efficeint way
In [1]: nums = [1, 8, 2, 23, 7, -4, 18, 23, 42, 37, 2]
In [2]: import heapq
In [3]: print (heapq.nlargest(3,nums))
[42, 37, 23]
In [4]: print (heapq.nsmallest(3,nums))
[-4, 1, 2]
In [5]: portfolio =[{'name':"Ajay",'age':24,'sex':'Male'},
...: {'name':"Cam",'age':23,'sex':'Male'},
...: {'name':"Cyber",'age':16,'sex':'Female'}]
In [6]:
In [6]: youngest=heapq.nsmallest(1,portfolio,key=lambda s: s['age'])
In [7]: youngest
Out[7]: [{'age': 16, 'name': 'Cyber', 'sex': 'Female'}]
In [8]: youngest=heapq.nsmallest(2,portfolio,key=lambda s: s['age'])
In [9]: youngest
Out[9]:
[{'age': 16, 'name': 'Cyber', 'sex': 'Female'},
{'age': 23, 'name': 'Cam', 'sex': 'Male'}]
In [10]: eldest=heapq.nlargest(1,portfolio,key=lambda s: s['age'])
In [11]: eldest
Out[11]: [{'age': 24, 'name': 'Ajay', 'sex': 'Male'}]
"""
A heap is a tree-like data structure where the child nodes have a sort-order relationship
with the parents. Binary heaps can be represented using a list or an array organized
so that the children of element N are at positions 2*N+1 and 2*N+2 (for zero-based
indexes). This layout makes it possible to rearrange heaps in place, so it is not necessary
to reallocate as much memory when adding or removing items.
A max-heap ensures that the parent is larger than or equal to both of its children.
A min-heap requires that the parent be less than or equal to its children. Python’s heapq
module implements a min-heap.
Check http://visualgo.net/heap.html(max heap implementation)
minheap --> https://www.cs.usfca.edu/~galles/visualization/Heap.html
http://kanaka.github.io/rbt_cfs/trees.html
check heapify,heappush,heapop from the standard library
Practical use of heapq
http://stackoverflow.com/questions/8627109/what-would-you-use-the-heapq-python-module-for-in-real-life
https://docs.python.org/3/library/heapq.html
"""
#To understand heap start with an empty list
>>> import heapq
>>> l=[1,9,2,4,8,5,6] # (L is llooking like 1)
>>> l=[]
>>> heapq.heappush(l,1)
>>> heapq.heappush(l,10)
>>> heapq.heappush(l,4)
>>> heapq.heappush(l,6)
>>> heapq.heappush(l,8)
>>> l
[1, 6, 4, 10, 8]
>>> heapq.heappop(l)
1
>>> l
[4, 6, 8, 10]
>>> heapq.heappushpop(l,83)
4
>>> l
[6, 10, 8, 83]
Learn python for fun.The popular blog with questions and answers to the python.Solutions to facebookhackercup,codejam,codechef.The fun way to learn python with me.Building some cool apps.Sunday, May 3, 2015
Basic usage of heapq in python
Labels:
ds-and-algo,
python
Subscribe to:
Post Comments (Atom)
No comments:
Post a Comment