logo

Funció de cap i cua a Python

A Python, el cap () La funció s'utilitza normalment per recuperar la primera n elements d'una llista o a DataFrame .

funció head() per a llistes

A Python, podeu utilitzar el cap () funció amb llistes per recuperar els primers n elements de la llista. La funció head() no és una funció integrada a Python, però podeu crear fàcilment la vostra pròpia funció per aconseguir la mateixa funcionalitat. Aquí teniu un exemple:

 def head(lst, n): return lst[:n] 

Aquesta funció pren una llista lst i un enter n com a entrada i retorna els primers n elements de la llista. Aquí teniu un exemple d'ús:

 my_list = [1, 2, 3, 4, 5] print(head(my_list, 3)) # prints [1, 2, 3] 

funció head() per a DataFrames

A Python, també podeu utilitzar el cap () funció amb pandas DataFrames per recuperar les primeres n files d'un DataFrame. La funció head() és una funció integrada a la biblioteca pandas, de manera que no cal que creeu la vostra pròpia funció. Aquí teniu un exemple:

slf4j vs log4j
 import pandas as pd # create a sample DataFrame df = pd.DataFrame({ 'name': ['Alice', 'Bob', 'Charlie', 'David', 'Emily'], 'age': [25, 30, 35, 40, 45] }) # use the head() function to retrieve the first 3 rows of the DataFrame print(df.head(3)) 

Sortida:

 name age 0 Alice 25 1 Bob 30 2 Charlie 35 

En aquest exemple, hem creat un DataFrame amb dues columnes ( nom i edat ) i cinc files. Després d'això, vam utilitzar el cap () funció per recuperar les tres primeres files del DataFrame.

A Python, la funció de cua s'utilitza per extreure un nombre especificat d'elements del final d'una seqüència, com ara una llista o una cadena. La seqüència resultant conté només els elements al final de la seqüència original, fins al nombre especificat d'elements.

Aquí teniu un exemple d'ús de la funció de cua en una llista en Python:

 def tail(seq, n): return seq[-n:] my_list = [1, 2, 3, 4, 5] result = tail(my_list, 3) print(result) # [3, 4, 5] 

Sortida:

 [3, 4, 5] 

En aquest exemple, la funció de cua pren dos arguments: seq i n . seq és el seqüència (en aquest cas, una llista) de la qual volem extreure els elements al final, i n és el nombre d'elements volem extreure.

tutorial de pyspark

Després d'això, la funció utilitza la sintaxi de tall de Python per extreure els darrers n elements de la seqüència. El -n: notació significa 'agafar els darrers n elements' , tan seq[-n:] retorna una llista nova que conté només els tres últims elements de la meva_llista.

Tingueu en compte que la llista original la meva_llista no es modifica per la cua funció. En canvi, la funció retorna una llista nova que només conté el nombre especificat d'elements del final de la llista original.

Aquesta funció de cua es pot utilitzar de diverses maneres:

caràcter a int java

Per exemple, quan necessitem extreure N caràcters des del final d'una cadena, que es pot implementar de la següent manera:

 def tail(string, n): return string[-n:] my_string = 'Hello, world!' result = tail(my_string, 6) print(result) 

Sortida:

 world! 

En aquest exemple, cua(la meva_cadena, 6) retorna els sis últims caràcters de la cadena, que són 'món!'.

Exemple 2: extreure les últimes N línies d'un fitxer

Suposem que teniu un fitxer de text i voleu extreure'n les últimes n línies. Podeu utilitzar la funció de cua juntament amb la funció oberta integrada de Python per llegir el fitxer i extreure les últimes n línies:

 def tail(file_path, n): with open(file_path, 'r') as f: lines = f.readlines() return lines[-n:] file_path = 'my_file.txt' result = tail(file_path, 5) print(result) 

Sortida:

tipus de retorn en java
 ['Line 6
', 'Line 7
', 'Line 8
', 'Line 9
', 'Line 10
'] 

En aquest exemple, cua(camí_fitxer, 5) llegeix el fitxer a la ruta especificada i retorna les últimes cinc línies del fitxer.

Exemple 3: Ús de Tail amb altres funcions

La funció de cua es pot combinar amb altres funcions de Python per realitzar operacions més complexes. Per exemple, suposem que teniu una llista de nombres i voleu calcular la mitjana mòbil dels darrers n nombres. Podeu utilitzar la funció de cua juntament amb la funció de suma integrada de Python per calcular la suma dels darrers n nombres:

 def tail(lst, n): return lst[-n:] def moving_average(lst, n): last_n = tail(lst, n) return sum(last_n) / n my_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] result = moving_average(my_list, 3) print(result) 

Sortida:

 8.0 

En aquest exemple, mitjana_mòbil(la meva_llista, 3) utilitza la funció de cua per extreure els tres últims nombres de la llista i després calcula la seva mitjana mitjançant la funció suma.