Python Enumerate. Wykorzystanie w analizie danych

08.2024 | Python | Enumerate | Analiza Danych

Zobaczymy, jak funkcja enumerate, dostępna w Pythonie może stać się naszym sprzymierzeńcem w zadanich związanych z analizą danych. Nie tylko uprości nasz kod, ale także sprawi, że będzie on bardziej czytelny i wydajny.

Zobaczmy zatem, czym jest enumerate …

1. Co to jest enumerate?

W skrócie, enumerate to wbudowana funkcja w Pythonie, która pozwala nam iterować po sekwencji (np. liście, krotce czy stringu) i jednocześnie otrzymywać zarówno indeks, jak i wartość każdego elementu. To tak, jakbyśmy mieli dwa liczniki w jednym - jeden dla indeksu, a drugi dla wartości.

2. Składnia funkcji enumerate

Składnia funkcji enumerate jest następująca:

enumerate(iterable, start=0)

• iterable: Obiekt, po którym chcesz iterować (np. lista, krotka, ciąg znaków).
• start (opcjonalny): Wartość początkowa licznika indeksów.

3. Przykład wykorzystania

Zacznijmy od prostego przykładu:

owoce = ['jabłko', 'banan', 'pomarańcza']

for indeks, owoc in enumerate(owoce):
    print(f"Owoc numer {indeks + 1} to {owoc}")

W tym przykładzie, enumerate przechodzi przez listę owoce i dla każdego elementu zwraca jego indeks (zaczynając od 0) oraz wartość. Dzięki temu możemy łatwo wyświetlić numer każdego owocu oraz jego nazwę.

4. Analiza danych - praktyczne zastosowania

Teraz przejdźmy do tego, jak enumerate może być przydatne w analizie danych.

Przykład 1. Śledzenie pozycji w zbiorze danych

Załóżmy, że analizowany zbiór danych zawieraja informacje o sprzedaży produktów w naszym sklepie internetowym. Chcemy znaleźć produkt, który sprzedał się najlepiej, oraz jego pozycję na liście.

W tym celu wykorzystujemy enumerate:

sprzedaz = [120, 85, 200, 150, 90]

najlepsza_sprzedaz = max(sprzedaz)
for indeks, wartosc in enumerate(sprzedaz):
    if wartosc == najlepsza_sprzedaz:
        pozycja = indeks + 1
        break

print(f"Produkt o pozycji {pozycja} sprzedał się najlepiej z wynikiem {najlepsza_sprzedaz} sztuk.")

Przykład 2. Przetwarzanie danych w parach

Czasami może zajść potrzeba porównania kolejnych elementów w zbiorze danych. enumerate może nam w tym pomóc:

ceny = [10.5, 12.0, 9.8, 11.2]

for indeks, cena in enumerate(ceny):
    if indeks > 0:
        zmiana = cena - ceny[indeks - 1]
        print(f"Zmiana ceny od poprzedniego dnia: {zmiana}")

Przykład 3. Tworzenie słowników z indeksami

Jeśli chcemy szybko stworzyć słownik, w którym klucze to indeksy, a wartości to elementy z sekwencji, enumerate jest idealnym narzędziem:

kolumny = ['nazwa', 'cena', 'ilosc']

slownik_kolumn = {indeks: kolumna for indeks, kolumna in enumerate(kolumny)}
print(slownik_kolumn)

Podsumowanie

Funkcja enumerate to pożyteczne narzędzie, które może znacznie ułatwić pracę z danymi w Pythonie. Dzięki tej metodzie możemy łatwo śledzić pozycje elementów, przetwarzać dane w parach oraz tworzyć słowniki z indeksami.

Omawiany artykuł pokazał nam, jak enumerate może stać się pożytecznym sprzymierzeńcem w procesie analizy danych.