Tính hệ số tương quan Pearson
Như đã đề cập trong video, hệ số tương quan Pearson (còn gọi là Pearson r) thường dễ diễn giải hơn hiệp phương sai. Nó được tính bằng hàm np.corrcoef(). Tương tự np.cov(), hàm này nhận hai mảng làm đối số và trả về một mảng 2D. Các phần tử [0,0] và [1,1] chắc chắn bằng 1 (bạn có đoán được vì sao không?), và giá trị chúng ta cần là phần tử [0,1].
Trong bài này, bạn sẽ viết một hàm pearson_r(x, y) nhận vào hai mảng và trả về hệ số tương quan Pearson. Sau đó, bạn sẽ dùng hàm này để tính hệ số cho chiều dài và chiều rộng cánh hoa của I. versicolor.
Chúng tôi cũng đính kèm lại biểu đồ phân tán bạn đã tạo ở bài trước để gợi nhớ mối quan hệ giữa chiều rộng và chiều dài cánh hoa.
Bài tập này là một phần của khóa học
Tư duy Thống kê với Python (Phần 1)
Hướng dẫn bài tập
- Định nghĩa một hàm với chữ ký
pearson_r(x, y).- Dùng
np.corrcoef()để tính ma trận tương quan củaxvày(truyền chúng vàonp.corrcoef()theo đúng thứ tự đó). - Hàm trả về phần tử
[0,1]của ma trận tương quan.
- Dùng
- Tính hệ số tương quan Pearson giữa dữ liệu trong các mảng
versicolor_petal_lengthvàversicolor_petal_width. Gán kết quả chor. - In kết quả.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
def ____(____, ____):
"""Compute Pearson correlation coefficient between two arrays."""
# Compute correlation matrix: corr_mat
# Return entry [0,1]
return corr_mat[0,1]
# Compute Pearson correlation coefficient for I. versicolor: r
# Print the result