
    Q/Phy                     |    d dl Z d dlmZ d dlmZ d dlmZ d dlmZm	Z	 e	j
        dededee         fd	            ZdS )
    N)List)	DataFrame)Settings)Sample
get_sampleconfigdfreturnc                    g }t          |                    d                    dk    r|S | j        j        }|dk    rJ|                    t	          d|                    |                                          d                     | j        j        }|dk    rt          j	        d           | j        j
        }|dk    rt          j	        d           |S )zObtains a sample from head and tail of the DataFrame

    Args:
        config: Settings object
        df: the spark DataFrame

    Returns:
        a list of Sample objects
       r   headz
First rows)iddatanamez]tail sample not implemented for spark. Set config.samples.n_tail to 0 to disable this warningzarandom sample not implemented for spark. Set config.samples.n_random to 0 to disable this warning)lenr   samplesappendr   limittoPandastailwarningswarnrandom)r   r	   r   n_headn_tailn_randoms         h/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/ydata_profiling/model/spark/sample_spark.pyget_sample_sparkr   
   s     G
2771::!^ Fzzf288F#3#3#<#<#>#>\RRR	
 	
 	
 ^ Fzzk	
 	
 	
 ~$H!||o	
 	
 	
 N    )r   typingr   pyspark.sql.dataframer   ydata_profiling.configr   ydata_profiling.model.sampler   r   registerr    r   r   <module>r&      s           + + + + + + + + + + + + ; ; ; ; ; ; ; ;  X  9  f            r   