accept numpy.array in HeatMap

Author: Martin Journois

folium/plugins/heat_map.py

@@ -20,8 +20,9 @@ def __init__(self, data, name=None,
 
         Parameters
         ----------
-        data : list of points of the form [lat, lng, weight]
-            The point you want to plot.
+        data : list of points of the form [lat, lng] or [lat, lng, weight]
+            The points you want to plot.
+            You can alos provide a numpy.array of shape (n,2) or (n,3).
         name : str
             The name of the layer that will be created.
         min_opacity  : default 1.
@@ -42,7 +43,7 @@ def __init__(self, data, name=None,
         self._name = 'HeatMap'
         self.tile_name = name if name is not None else self.get_name()
 
-        self.data= data
+        self.data = [[x for x in line] for line in data]
         self.min_opacity = min_opacity
         self.max_zoom = max_zoom
         self.max_val = max_val