Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Python Pandas - 각 그룹의 행 수 계산

<시간/>

group.size()를 사용하여 각 그룹의 행 수를 계산합니다. 필요한 라이브러리 가져오기 -

import pandas as pd

DataFrame 생성 -

dataFrame = pd.DataFrame({'Product Category': ['Computer', 'Mobile Phone', 'Electronics', 'Electronics', 'Computer', 'Mobile Phone'],'Quantity': [10, 50, 10, 20, 25, 50],'Product Name': ['Keyboard', 'Charger', 'SmartTV', 'Camera', 'Graphic Card', 'Earphone']
})

열별로 그룹화 -

dataFrame.groupby(["Product Category", "Quantity"])

이제 그룹 크기를 계산하여 각 그룹의 행 수를 구합니다.

예시

다음은 전체 코드입니다 -

import pandas as pd

# create a dataframe
dataFrame = pd.DataFrame({'Product Category': ['Computer', 'Mobile Phone', 'Electronics', 'Electronics', 'Computer', 'Mobile Phone'],'Quantity': [10, 50, 10, 20, 25, 50],'Product Name': ['Keyboard', 'Charger', 'SmartTV', 'Camera', 'Graphic Card', 'Earphone']
})

# dataframe
print"Dataframe...\n",dataFrame

# grouping columns
new_group = dataFrame.groupby(["Product Category", "Quantity"])

# group size
new_group = new_group.size()

# dataframe
print"\nUpdated Dataframe...\n",new_group

출력

이것은 다음과 같은 출력을 생성합니다 -

Dataframe...
   Product Category   Product Name   Quantity
0          Computer       Keyboard         10
1      Mobile Phone        Charger         50
2       Electronics        SmartTV         10
3       Electronics         Camera         20
4          Computer   Graphic Card         25
5      Mobile Phone       Earphone         50

Updated Dataframe...
Product Category   Quantity
Computer           10        1
                   25        1
Electronics        10        1
                   20        1
Mobile Phone       50        2
dtype: int64