Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

파이썬 Pandas DataFrame에 0으로 채워진 열 추가하는 방법

Pandas DataFrame에 0 열 추가하기

Pandas DataFrame에 모든 값이 0인 새로운 열을 추가하는 방법은 매우 간단합니다. 대괄호([]) 안에 새 열 이름을 지정하고 값으로 0을 할당하기만 하면 됩니다. 이렇게 하면 DataFrame의 모든 행에 대해 해당 열의 값이 자동으로 0으로 설정됩니다.

먼저 필요한 라이브러리를 임포트합니다.

import pandas as pd

3개의 열(Student, Result, Roll Number)을 가진 DataFrame을 생성해 보겠습니다.

dataFrame = pd.DataFrame(
    {
        "Student": ['Jack', 'Robin', 'Ted', 'Marc', 'Scarlett', 'Kat', 'John'],
        "Result": ['Pass', 'Fail', 'Pass', 'Fail', 'Pass', 'Pass', 'Pass'],
        "Roll Number": [5, 10, 3, 8, 2, 9, 6]
    }
)

이제 대괄호 표기법을 사용하여 0으로 채워진 새로운 열을 추가합니다.

dataFrame['ZeroColumn'] = 0

새 열 이름 'ZeroColumn'을 대괄호 안에 지정하고 0을 할당하면, 기존 DataFrame에 해당 열이 추가되며 모든 행의 값이 0으로 채워집니다.

전체 예제 코드

다음은 위 과정을 하나로 정리한 완전한 코드입니다.

import pandas as pd

# DataFrame 생성
dataFrame = pd.DataFrame(
    {
        "Student": ['Jack', 'Robin', 'Ted', 'Marc', 'Scarlett', 'Kat', 'John'],
        "Result": ['Pass', 'Fail', 'Pass', 'Fail', 'Pass', 'Pass', 'Pass'],
        "Roll Number": [5, 10, 3, 8, 2, 9, 6]
    }
)

print("DataFrame ...\n", dataFrame)

dataFrame['ZeroColumn'] = 0

print("\n업데이트된 DataFrame ...\n", dataFrame)

실행 결과

위 코드를 실행하면 다음과 같은 출력 결과를 얻을 수 있습니다.

DataFrame ...
    Result  Roll Number  Student
0    Pass            5     Jack
1    Fail           10    Robin
2    Pass            3      Ted
3    Fail            8     Marc
4    Pass            2  Scarlett
5    Pass            9      Kat
6    Pass            6     John

업데이트된 DataFrame ...
    Result  Roll Number  Student  ZeroColumn
0    Pass            5     Jack           0
1    Fail           10    Robin           0
2    Pass            3      Ted           0
3    Fail            8     Marc           0
4    Pass            2  Scarlett           0
5    Pass            9      Kat           0
6    Pass            6     John           0

추가 팁: 데이터 타입 지정하기

특정 데이터 타입을 명시적으로 지정하고 싶다면 NumPy를 활용할 수도 있습니다.

import numpy as np
dataFrame['ZeroColumn'] = np.zeros(len(dataFrame), dtype=int)

실수형(float) 0을 원한다면 dataFrame['ZeroColumn'] = 0.0처럼 작성하면 됩니다. 이처럼 스칼라 값을 할당하는 방식은 DataFrame 초기화나 계산용 임시 열을 만들 때 유용하게 사용됩니다.