파이썬으로 데이터프레임에 행 추가하는 방법

[

헤딩 추가하기

헤딩 추가하기

데이터프레임에 행 추가하는 방법

데이터프레임은 파이썬에서 많이 사용되는 데이터 구조 중 하나로, 표 형태의 데이터를 다루는 데 유용합니다. 데이터프레임에 기존 행에 새로운 행을 추가하는 방법을 알아보겠습니다. 이를 위해 파이썬의 판다스(Pandas) 라이브러리를 사용할 것입니다.

1. DataFrame 생성하기

먼저, 데이터프레임을 생성해야 합니다. 다음과 같은 예제 데이터프레임을 사용하겠습니다.

import pandas as pd

data = {'Name': ['John', 'Emma', 'Mike'],
        'Age': [25, 28, 30],
        'City': ['Seoul', 'New York', 'London']}

df = pd.DataFrame(data)
print(df)

출력 결과는 다음과 같습니다.

   Name  Age      City
0  John   25     Seoul
1  Emma   28  New York
2  Mike   30    London

2. 새로운 행 추가하기

이제 새로운 행을 데이터프레임에 추가해 보겠습니다. loc 함수를 사용하여 행을 추가할 수 있습니다. 다음은 예제 코드입니다.

new_row = {'Name': 'Sarah', 'Age': 32, 'City': 'Paris'}
df = df.append(new_row, ignore_index=True)
print(df)

출력 결과는 다음과 같습니다.

   Name  Age      City
0  John   25     Seoul
1  Emma   28  New York
2  Mike   30    London
3  Sarah  32    Paris

위의 코드에서 append 함수를 사용하여 새로운 행을 추가했습니다. ignore_index=True 옵션을 추가하여 새로운 행의 인덱스를 재설정했습니다.

3. 여러 개의 행 추가하기

한 번에 여러 개의 행을 추가할 수도 있습니다. 다음은 여러 개의 행을 추가하는 예제 코드입니다.

new_rows = [{'Name': 'David', 'Age': 29, 'City': 'Berlin'},
            {'Name': 'Linda', 'Age': 26, 'City': 'Tokyo'}]
df = df.append(new_rows, ignore_index=True)
print(df)

출력 결과는 다음과 같습니다.

   Name  Age      City
0  John   25     Seoul
1  Emma   28  New York
2  Mike   30    London
3  Sarah  32    Paris
4  David  29    Berlin
5  Linda  26     Tokyo

4. 행 인덱스 추가하기

기존의 행 인덱스를 유지하면서 새로운 행을 추가할 수도 있습니다. 다음은 행 인덱스를 추가하는 예제 코드입니다.

new_row = {'Name': 'Sophia', 'Age': 27, 'City': 'Sydney'}
df = df.append(pd.Series(new_row, name=len(df)))
print(df)

출력 결과는 다음과 같습니다.

   Name  Age      City
0  John   25     Seoul
1  Emma   28  New York
2  Mike   30    London
3  Sarah  32    Paris
4  David  29    Berlin
5  Linda  26     Tokyo
6  Sophia 27    Sydney

위의 코드에서는 pd.Series 함수를 사용하여 새로운 행을 시리즈 형태로 생성한 후, name 매개 변수를 설정하여 새로운 행의 인덱스를 지정했습니다.

5. 예외 처리하기

만약, 행 추가 중에 예외 상황이 발생하는 경우에는 예외 처리를 해주어야 합니다. 예를 들어, 행의 개수나 열의 개수가 일치하지 않을 경우 오류가 발생할 수 있습니다. 이를 방지하기 위해 예외 처리를 추가하는 것이 좋습니다.

try:
    new_row = {'Name': 'Tom', 'Age': 31, 'City': 'Moscow', 'Country': 'Russia'}
    df = df.append(pd.Series(new_row, name=len(df)))
except ValueError as e:
    print(e)

위의 코드에서는 try와 except를 사용하여 ValueError 예외를 처리합니다. 행의 개수와 열의 개수가 일치하지 않을 때 append 함수에서 오류가 발생하고, 이를 except 블록에서 처리합니다.

---

이렇게 파이썬에서 데이터프레임에 행을 추가하는 방법을 알아보았습니다. 위의 단계별 예제 코드를 실행해보면서 데이터프레임에 새로운 행을 손쉽게 추가할 수 있습니다.