logo

Funcția cap și coadă în Python

În Python, cap() funcția este de obicei folosită pentru a prelua prima n articole dintr-o listă sau a DataFrame .

funcția head() pentru liste

În Python, puteți utiliza cap() funcția cu liste pentru a prelua primele n elemente din listă. Funcția head() nu este o funcție încorporată în Python, dar vă puteți crea cu ușurință propria funcție pentru a obține aceeași funcționalitate. Iată un exemplu:

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

Această funcție ia o listă lst și un întreg n ca intrare și returnează primele n elemente din listă. Iată un exemplu de utilizare:

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

funcția head() pentru DataFrames

În Python, puteți utiliza și cap() funcția cu Pandas DataFrames pentru a prelua primele n rânduri ale unui DataFrame. Funcția head() este o funcție încorporată în biblioteca panda, așa că nu trebuie să vă creați propria funcție. Iată un exemplu:

pagini server java
 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)) 

Ieșire:

 name age 0 Alice 25 1 Bob 30 2 Charlie 35 

În acest exemplu, am creat un DataFrame cu doua coloane ( Nume și vârstă ) și cinci rânduri. După aceea, am folosit cap() funcția pentru a prelua primele trei rânduri ale DataFrame.

În Python, funcția coadă este utilizată pentru a extrage un număr specificat de elemente de la sfârșitul unei secvențe, cum ar fi o listă sau un șir. Secvența rezultată conține doar elementele de la sfârșitul secvenței originale, până la numărul specificat de elemente.

Iată un exemplu de utilizare a funcției coadă pe o listă în Python:

java citește fișierul linie cu linie
 def tail(seq, n): return seq[-n:] my_list = [1, 2, 3, 4, 5] result = tail(my_list, 3) print(result) # [3, 4, 5] 

Ieșire:

 [3, 4, 5] 

În acest exemplu, funcția coadă ia două argumente: secv și n . secv este secvenţă (în acest caz, o listă) din care dorim să extragem elementele de la sfârșit, și n este numărul de articole vrem să extragem.

După aceea, funcția folosește sintaxa de feliere a lui Python pentru a extrage ultimele n elemente ale secvenței. The -n: notație mijloace „ia ultimele n articole” , asa de secv[-n:] returnează o nouă listă care conține doar ultimele trei elemente din my_list.

Rețineți că lista originală lista mea nu este modificat de către coadă funcţie. În schimb, funcția returnează o listă nouă care conține doar numărul specificat de articole de la sfârșitul listei originale.

Această funcție de coadă poate fi utilizată în mai multe moduri:

De exemplu, când trebuie să extragem N caractere de la sfârșitul unui șir, care poate fi implementat după cum urmează:

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

Ieșire:

încercați structura de date
 world! 

În acest exemplu, coada(șirul_meu, 6) returnează ultimele șase caractere ale șirului, care sunt 'lume!'.

Exemplul 2: Extragerea ultimelor N linii dintr-un fișier

Să presupunem că aveți un fișier text și doriți să extrageți ultimele n linii din acesta. Puteți utiliza funcția de coadă împreună cu funcția de deschidere încorporată a lui Python pentru a citi fișierul și a extrage ultimele n linii:

 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) 

Ieșire:

 ['Line 6
', 'Line 7
', 'Line 8
', 'Line 9
', 'Line 10
'] 

În acest exemplu, coada(cale_fișier, 5) citește fișierul la calea specificată și returnează ultimele cinci rânduri ale fișierului.

Exemplul 3: Utilizarea Cozii cu alte funcții

Funcția coadă poate fi combinată cu alte funcții Python pentru a efectua operații mai complexe. De exemplu, să presupunem că aveți o listă de numere și doriți să calculați media mobilă a ultimelor n numere. Puteți utiliza funcția coadă împreună cu funcția de sumă încorporată din Python pentru a calcula suma ultimelor n numere:

css pentru împachetare text
 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) 

Ieșire:

 8.0 

În acest exemplu, media_mobilă(lista_mea, 3) folosește funcția coadă pentru a extrage ultimele trei numere din listă, apoi calculează media lor folosind funcția sumă.