[PYTHON/PANDAS] DataFrame 클래스 : [] 연산자를 사용해 컬럼에 대한 Series 객체 구하기

■ DataFrame 클래스에서 [] 연산자를 사용해 컬럼에 대한 Series 객체를 구하는 방법을 보여준다.

▶ main.py


import pandas as pd
import numpy  as np

np.random.seed(0)

category1List       = ["수익", "비용"]
category2Dictionary = {"수익" : ["영업수익", "영업외수익"], "비용" : ["영업비용", "영업외비용"]}
category3Dictionary = {
    "영업수익"   : ["제품매출", "용역매출"  ],
    "영업외수익" : ["이자수익", "배당금수익"],
    "영업비용"   : ["인건비"  , "재료비"    ],
    "영업외비용" : ["이자비용", "기부금"    ]
}

sourceList = []

def generateAccountCode(대분류, 중분류, 소분류):
    return f"{대분류[:1]}{중분류[:1]}{소분류[:1]}"

for category1 in category1List:
    for category2 in category2Dictionary[category1]:
        for category3 in category3Dictionary[category2]:
            accountCode = generateAccountCode(category1, category2, category3)
            for month in range(1, 13):
                targetAmount = np.random.randint(10000, 100000)
                actualAmount = np.random.randint(8000 , 120000)
                sourceList.append([category1, category2, category3, accountCode, f"2023-{month:02d}", targetAmount, actualAmount])

dataFrame = pd.DataFrame(sourceList, columns = ["대분류", "중분류", "소분류", "계정코드", "해당월", "목표금액", "실적금액"])

for columnName in dataFrame.columns:
    series = dataFrame[columnName]
    print("-" * 100)
    print(columnName)
    for value in series:
        print(value, end = ",", )
    print()

import pandas as pd

import numpy as np

np.random.seed(0)

category1List = ["수익", "비용"]

category2Dictionary = {"수익" : ["영업수익", "영업외수익"], "비용" : ["영업비용", "영업외비용"]}

category3Dictionary = {

"영업수익" : ["제품매출", "용역매출" ],

"영업외수익" : ["이자수익", "배당금수익"],

"영업비용" : ["인건비" , "재료비" ],

"영업외비용" : ["이자비용", "기부금" ]

}

sourceList = []

def generateAccountCode(대분류, 중분류, 소분류):

return f"{대분류[:1]}{중분류[:1]}{소분류[:1]}"

for category1 in category1List:

for category2 in category2Dictionary[category1]:

for category3 in category3Dictionary[category2]:

accountCode = generateAccountCode(category1, category2, category3)

for month in range(1, 13):

targetAmount = np.random.randint(10000, 100000)

actualAmount = np.random.randint(8000 , 120000)

sourceList.append([category1, category2, category3, accountCode, f"2023-{month:02d}", targetAmount, actualAmount])

dataFrame = pd.DataFrame(sourceList, columns = ["대분류", "중분류", "소분류", "계정코드", "해당월", "목표금액", "실적금액"])

for columnName in dataFrame.columns:

series = dataFrame[columnName]

print("-" * 100)

print(columnName)

for value in series:

print(value, end = ",", )

print()

▶ requirements.txt


numpy==2.1.2
pandas==2.2.3
python-dateutil==2.9.0.post0
pytz==2024.2
six==1.16.0
tzdata==2024.2

numpy==2.1.2

pandas==2.2.3

python-dateutil==2.9.0.post0

pytz==2024.2

six==1.16.0

tzdata==2024.2

※ pip install pandas 명령을 실행했다.