logo

Huvud- och svansfunktion i Python

I Python, den huvud() funktionen används vanligtvis för att hämta den första n objekt från en lista eller en DataFrame .

head() funktion för listor

I Python kan du använda huvud() funktion med listor för att hämta de första n objekten i listan. Head()-funktionen är inte en inbyggd funktion i Python, men du kan enkelt skapa din egen funktion för att uppnå samma funktionalitet. Här är ett exempel:

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

Denna funktion tar en lista lst och ett heltal n som indata och returnerar de första n objekten i listan. Här är ett exempel på användning:

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

head() funktion för DataFrames

I Python kan du också använda huvud() funktion med pandas DataFrames för att hämta de första n raderna i en DataFrame. Head()-funktionen är en inbyggd funktion i pandas-biblioteket, så du behöver inte skapa din egen funktion. Här är ett exempel:

vad är desktop.ini
 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)) 

Produktion:

 name age 0 Alice 25 1 Bob 30 2 Charlie 35 

I det här exemplet skapade vi en DataFrame med två kolumner ( namn och ålder ) och fem rader. Efter det använde vi huvud() funktion för att hämta de tre första raderna i DataFrame.

I Python används svansfunktionen för att extrahera ett specificerat antal objekt från slutet av en sekvens, till exempel en lista eller en sträng. Den resulterande sekvensen innehåller endast objekten i slutet av den ursprungliga sekvensen, upp till det angivna antalet objekt.

enkelt java-program

Här är ett exempel på hur man använder svansfunktionen på en lista i 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] 

Produktion:

 [3, 4, 5] 

I det här exemplet tar svansfunktionen två argument: seq och n . seq är sekvens (i det här fallet en lista) från vilken vi vill extrahera objekten i slutet, och n är antal artiklar vi vill utvinna.

Därefter använder funktionen Pythons skivsyntax för att extrahera de sista n objekten i sekvensen. De -n: notation betyder 'ta de sista n objekten' , alltså seq[-n:] returnerar en ny lista som endast innehåller de tre sista objekten i min_lista.

räkna distinkta sql

Observera att den ursprungliga listan min lista är inte modifierad av svans fungera. Istället returnerar funktionen en ny lista som endast innehåller det angivna antalet objekt från slutet av den ursprungliga listan.

Denna svansfunktion kan användas på flera sätt:

Till exempel när vi behöver extrahera N tecken från slutet av en sträng, som kan implementeras enligt följande:

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

Produktion:

 world! 

I det här exemplet, svans(min_sträng, 6) returnerar de sista sex tecknen i strängen, som är 'värld!'.

konvertera ett datum till en sträng

Exempel 2: Extrahera de sista N raderna från en fil

Anta att du har en textfil och vill extrahera de sista n raderna från den. Du kan använda svansfunktionen tillsammans med Pythons inbyggda öppna funktion för att läsa filen och extrahera de sista n raderna:

 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) 

Produktion:

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

I det här exemplet, tail(fil_sökväg, 5) läser filen på den angivna sökvägen och returnerar de sista fem raderna i filen.

Exempel 3: Använda Tail med andra funktioner

Svansfunktionen kan kombineras med andra Python-funktioner för att utföra mer komplexa operationer. Anta till exempel att du har en lista med siffror och att du vill beräkna det glidande medelvärdet av de sista n talen. Du kan använda svansfunktionen tillsammans med Pythons inbyggda summafunktion för att beräkna summan av de sista n talen:

 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) 

Produktion:

 8.0 

I det här exemplet, moving_average(min_lista, 3) använder svansfunktionen för att extrahera de tre sista siffrorna från listan, och beräknar sedan deras medelvärde med hjälp av summafunktionen.