logo

Funkcja głowy i ogona w Pythonie

W Pythonie głowa() Funkcja jest zwykle używana do pobrania pierwszej N elementy z listy lub a Ramka danych .

funkcja head() dla list

W Pythonie możesz użyć głowa() funkcja z listami, aby pobrać pierwszych n elementów z listy. Funkcja head() nie jest funkcją wbudowaną w Pythonie, ale możesz łatwo stworzyć własną funkcję, aby osiągnąć tę samą funkcjonalność. Oto przykład:

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

Ta funkcja pobiera listę lst i liczbę całkowitą n jako dane wejściowe i zwraca pierwszych n elementów na liście. Oto przykładowe użycie:

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

funkcja head() dla ramek danych

W Pythonie możesz także użyć metody głowa() funkcja z pandami DataFrames, aby pobrać pierwsze n wierszy DataFrame. Funkcja head() jest funkcją wbudowaną w bibliotekę pand, więc nie musisz tworzyć własnej funkcji. Oto przykład:

oś serwera sql
 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)) 

Wyjście:

 name age 0 Alice 25 1 Bob 30 2 Charlie 35 

W tym przykładzie utworzyliśmy plik Ramka danych z dwiema kolumnami ( nazwa I wiek ) i pięć rzędów. Następnie skorzystaliśmy z głowa() funkcja do pobrania pierwszych trzech wierszy ramki DataFrame.

W Pythonie funkcja tail służy do wyodrębnienia określonej liczby elementów z końca sekwencji, takich jak lista lub ciąg znaków. Wynikowa sekwencja zawiera tylko elementy znajdujące się na końcu oryginalnej sekwencji, aż do określonej liczby elementów.

Oto przykład użycia funkcji ogona na liście w Pythonie:

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

Wyjście:

 [3, 4, 5] 

W tym przykładzie funkcja ogona przyjmuje dwa argumenty: nast I N . nast jest sekwencja (w tym przypadku listę), z której na końcu chcemy wyodrębnić elementy, oraz N jest Liczba przedmiotów chcemy wydobyć.

Następnie funkcja wykorzystuje składnię Pythona do wycinania, aby wyodrębnić n ostatnich elementów sekwencji. The -n: notacja oznacza „weź ostatnich n przedmiotów” , Więc sekwencja[-n:] zwraca nową listę zawierającą tylko trzy ostatnie elementy z my_list.

Należy pamiętać, że oryginalna lista moja lista nie jest modyfikowany przez ogon funkcjonować. Zamiast tego funkcja zwraca nową listę zawierającą tylko określoną liczbę elementów z końca oryginalnej listy.

Tej funkcji ogona można używać na kilka sposobów:

Na przykład, gdy musimy wyodrębnić N znaków z końca ciągu, co można zaimplementować w następujący sposób:

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

Wyjście:

klucz podstawowy klucz złożony
 world! 

W tym przykładzie ogon(my_string, 6) zwraca sześć ostatnich znaków ciągu, czyli 'świat!'.

Przykład 2: Wyodrębnianie ostatnich N linii z pliku

Załóżmy, że masz plik tekstowy i chcesz wyodrębnić z niego n ostatnich wierszy. Możesz użyć funkcji tail wraz z wbudowaną w Pythonie funkcją open, aby odczytać plik i wyodrębnić n ostatnich 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) 

Wyjście:

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

W tym przykładzie ogon(ścieżka_pliku, 5) odczytuje plik pod określoną ścieżką i zwraca pięć ostatnich wierszy pliku.

Przykład 3: Używanie Tail z innymi funkcjami

Funkcję ogona można łączyć z innymi funkcjami Pythona, aby wykonywać bardziej złożone operacje. Załóżmy na przykład, że masz listę liczb i chcesz obliczyć średnią ruchomą ostatnich n liczb. Możesz użyć funkcji ogona wraz z wbudowaną funkcją sumy Pythona, aby obliczyć sumę ostatnich n liczb:

macierze w programowaniu w języku C
 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) 

Wyjście:

 8.0 

W tym przykładzie średnia ruchoma (moja_lista, 3) używa funkcji ogona do wyodrębnienia trzech ostatnich liczb z listy, a następnie oblicza ich średnią za pomocą funkcji sumy.