[Python] Numpy와 Pandas를 사용한 데이터셋 생성

Numpy를 사용해 2차원 데이터 배열 생성

import numpy as np

# 먼저 arr를 np.empty를 통해 초기화합니다.
arr = np.empty((0,3), int)

# 배열을 추가합니다.
# 추가하는 배열의 요소수는 초기화했던 길이와 동일해야합니다.
# axis = 0은 행으로 추가한다는 뜻입니다.
arr = np.append(arr, np.array([[1, 2, 3]]), axis=0)
arr = np.append(arr, np.array([[4, 5, 0]]), axis=0)

print(arr)
#[[1 2 3]
# [4 5 0]]

Pandas를 사용해 Numpy 배열을 DateFrame 형식으로 변환

import pandas as pd

df = pd.DataFrame(arr,columns=['first','second','third'])

print(df)
#     first  second  third
#0      1       2      3
#1      4       5      0