
    |Le
                       d Z ddlmZ edk    rddlmZmZmZ ddlm	Z
 ddlZddlZddlZ	 ddlZn&# e$ r  ed            ej        d           Y nw xY wd	 Z ej         ej        e                    Z ej        ed
          Z ej        e          s ee           d Z ej        ed          Z ej        e          s ed           dZ ee            ej        d          Z  ed           e j!        e j"        dk     e j"        dk    z  e j#        dk     z  e j#        dk    z  e j$        dk    z  e j$        dk     z  e j%        dk    z  e j%        dk     z           &                                Z  ed            ee dd            ee dd           e '                    dddddd            e (                    ed!"            ed            ed#           dS dS )$z%Download data needed for the examples    )annotations__main__)pathmakedirsremove)barNzADownload script required requests package: conda install requests   c                   t          j        | d          }t          j        |           d         }t	          |d          5 }t          |j                            d                    }t          |                    d          |dz  dz             D ]-}|r)|	                    |           |
                                 .	 d d d            d S # 1 swxY w Y   d S )	NT)streamr	   wbzcontent-lengthi   )
chunk_size)expected_size)requestsgetr   splitopenintheadersprogressbariter_contentwriteflush)urlroutput_pathftotal_lengthchunks         Nlib/python3.11/site-packages/datashader/examples/taxi_preprocessing_example.py_download_datasetr       s   LT***jooa(+t$$ 	qy}}-=>>??L$Q^^t^%D%DUabfUfjkTklll   GGENNNGGIII	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s   BCCCdatac                :   | |         }| |         }dt           j        z  dz  dz  }||z  dz  }t          j        t          j        d|z   t           j        z  dz                      t           j        dz  z  }||z  dz  }|| j        d d |f<   || j        d d |f<   d S )N   iRa g       @g     f@Z   g     v@)nppilogtanloc)dflat_namelng_namelatlngorigin_shiftmxmys           r   latlng_to_metersr2   $   s    ll25y7*S0<%'VBFBH-56677255=I,& qqq({ qqq({    znyc_taxi.csvzDownloading Taxi Data...zMhttps://storage.googleapis.com/tlc-trip-data/2015/yellow_tripdata_2015-01.csvzyellow_tripdata_2015-01.csvzFiltering Taxi Datag     pRgRgףp=
WD@gQkD@zReprojecting Taxi Datapickup_latitudepickup_longitudedropoff_latitudedropoff_longitudepickup_x	dropoff_xpickup_y	dropoff_y)r5   r7   r4   r6   T)columnsinplaceF)indexz
All data downloaded.))__doc__
__future__r   __name__osr   r   r   download_sample_datar   r   pandaspdnumpyr%   sysr   ImportErrorprintexitr    dirnamerealpath__file__examples_dirjoindata_direxistsr2   	taxi_pathr   read_csvr*   r)   r5   r7   r4   r6   copyrenameto_csv r3   r   <module>rX      s   + + " " " " " "z))))))))))777777JJJ   QRRR    4<h 7 788Ltyv..H4;x   ! ! ! 	(N33I4;y!! .()))- 	#R[677#$$$VR(61(613)F24 )F24 '%/	1
 '%/1 (502 (502 3 48466 	 	&'''.0BCCC/1DEEE
		zP[.8k[ [ 	 	  	  	  			)5	))),--- 
E
"#####M s   2  AA