Loc Python - изучаем язык программирования Python

В Python, функция loc() используется для фильтрации данных в объекте DataFrame с использованием меток строк и столбцов. Она возвращает новый объект DataFrame, содержащий только выбранные строки и столбцы.

Пример использования функции loc():

Предположим, у нас есть следующий DataFrame:

import pandas as pd
data = {
    'Name': ['John', 'Emma', 'Kate', 'Michael'],
    'Age': [25, 30, 28, 35],
    'City': ['New York', 'Paris', 'London', 'Tokyo']
}
df = pd.DataFrame(data)

Теперь давайте рассмотрим несколько примеров использования функции loc().

Пример 1: Выбор строк по условию

Мы можем использовать функцию loc() для выбора строк, удовлетворяющих определенному условию. Например, давайте выберем все строки, где возраст равен 30.

filtered_df = df.loc[df['Age'] == 30]
print(filtered_df)

Вывод:

   Name  Age   City
1  Emma   30  Paris

Пример 2: Выбор определенных столбцов

Мы можем использовать функцию loc() для выбора определенных столбцов из DataFrame. Например, давайте выберем только столбцы "Name" и "City".

filtered_df = df.loc[:, ['Name', 'City']]
print(filtered_df)

Вывод:

      Name      City
0     John  New York
1     Emma     Paris
2     Kate    London
3  Michael     Tokyo

Пример 3: Выбор строк и столбцов

Мы можем комбинировать условия для фильтрации строк и выбирать определенные столбцы. Например, давайте выберем строки, где возраст больше 25, и только столбцы "Name" и "Age".

filtered_df = df.loc[df['Age'] > 25, ['Name', 'Age']]
print(filtered_df)

Вывод:

      Name  Age
1     Emma   30
2     Kate   28
3  Michael   35

Таким образом, функция loc() позволяет эффективно фильтровать данные в объекте DataFrame в Python, используя метки строк и столбцов.