[DA와 ML] 06 회귀분석기법 미션 : X, Y 입력받기

이론 개념

[DA와 ML] 선형회귀법이란? 회귀분석이란?

해답

2. Numpy 사용하기

import numpy

def main():
    
    # 데이터 포인트의 구성 : X y로 구성되며 space로 구분. float을 포함
    # 문자열로 된 숫자를 실수로 변환시 float()사용 : str_val = "3.14" print(float(str_val)*2)
    (N, X, Y) = read_data()
    print(N)
    print(X)
    print(Y)

def read_data():
    # Implement here
    N = int(input()) # N : 데이터 갯수. N줄에 걸쳐 데이터 포인트들이 입력
    # X,Y는 빈리스트
    X = []
    Y = []
    for i in range(N) : 
        line = input().strip().split(" ") # N만큼 입력반복하여 X,Y의 빈리스트 채우기, X,Y 구분은 space
        X.append(float(line[0]))
        Y.append(float(line[1]))
    

    return (N, X, Y) # 데이터를 입력받아 N,X,Y 리턴

if __name__ == "__main__":
    main()