빅데이터분석기사/코드
[빅데이터분석기사] 실기 3유형 회귀모형(상관계수)(Python)
EveningPrimrose
2023. 6. 18. 22:53
반응형
회귀모형(상관계수)
iris에서 Sepal Length와 Sepal Width의 상관계수를 계산하고 소수 둘째자리까지 출력하시오
import pandas as pd
from sklearn.datasets import load_iris
# iris 데이터셋 로드
iris = load_iris()
df = pd.DataFrame(iris.data, column=iris.feature_names)
# Sepal Length와 Sepal Width의 상관계수 계산
correlation = df.corr()
result = correlation.loc['sepal length (cm)', 'sepal width (cm)']
print(round(result), 2))
반응형