Как перебирать группы фреймов данных в Python-Pandas?
В этой статье мы увидим, как мы можем перебирать группы, на которые разделен фрейм данных. Итак, давайте посмотрим, как можно выполнить эту задачу.
Во-первых, давайте создадим фрейм данных:
Code:
Python3
# import pandas libraryimport pandas as pd # dictionarydict = {"X": ["A", "B", "A", "B"], "Y": [1, 4, 3, 2]} # create a dataframedf = pd.DataFrame(dict) # show the dataframedf |
Выход:
Метод 1: Использование Dataframe.groupby () .
Эта функция используется для разделения данных на группы по некоторым критериям.
Example: we’ll simply iterate over all the groups created.
Python3
# import pandas libraryimport pandas as pd # dictionarydict = {"X": ["A", "B", "A", "B"], "Y": [1, 4, 3, 2]} # create a dataframedf = pd.DataFrame(dict) # group by "X" columngroups = df.groupby("X") for name, group in groups: print(name) print(group) print("
") |
Выход:
В приведенном выше примере мы сгруппировали на основе столбца «X». Поскольку в столбце «X» есть два разных значения, наш фрейм данных будет разделен на 2 группы. Затем цикл for будет выполняться 2 раза, так как группы номеров равны 2. «name» представляет имя группы, а «group» представляет собой фактический сгруппированный фрейм данных.
Метод 2: Совместное использование Dataframe.groupby () и Groupby_object.groups.keys () .
Метод Groupby_object.groups.keys () вернет ключи групп.
Example: we’ll iterate over the keys.
Python3
# import pandas libraryimport pandas as pd # dictionarydict = {"X": ["A", "B", "A", "B"], "Y": [1, 4, 3, 2]} # create a dataframedf = pd.DataFrame(dict) # group by "X" columngroups = df.groupby("X") # extract keys from groupskeys = groups.groups.keys() for i in keys: print(groups.get_group(i)) print("
") |
Выход:
В приведенном выше примере мы будем использовать функцию groups.get_group (), чтобы получить все группы. Сначала мы получим все ключи группы, а затем переберем их и затем вызовем метод get_group () для каждого ключа. Метод get_group () вернет группу, соответствующую ключу.
Внимание компьютерщик! Укрепите свои основы с помощью базового курса программирования Python и изучите основы.
Для начала подготовьтесь к собеседованию. Расширьте свои концепции структур данных с помощью курса Python DS. А чтобы начать свое путешествие по машинному обучению, присоединяйтесь к курсу Машинное обучение - базовый уровень.