Tương quan giữa hai cột số trong DataFrame Pandas

Chúng ta có thể dùng pandas.DataFrame.corr để tính tương quan cặp cột, loại trừ giá trị NULL. Hệ số tương quan chỉ cường độ liên kết tuyến tính giữa hai biến. Hệ số dao động từ -1 đến 1.

Để lấy tương quan giữa hai cột số trong DataFrame Pandas, thực hiện:

• Đặt kích thước hình và điều chỉnh padding.
• Tạo DataFrame Pandas dữ liệu hai chiều.
• So sánh giá trị hai cột và tính hệ số tương quan dùng col1.corr(col2).
• In hệ số tương quan ra console.
• Hiển thị hình bằng show().

Ví dụ

import pandas as pd
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
df = pd.DataFrame({'lab': [1, 2, 3], 'value': [3, 4, 5]})
col1 = df['lab']
col2 = df['value']
plt.plot(col1, col2)
print("Hệ số tương quan là: ", col1.corr(col2))
plt.show()

Kết quả

Sẽ tạo ra đầu ra sau:

Hệ số tương quan là: 1.0

Ở đây, hệ số tương quan là 1.0 cho thấy tương quan hoàn hảo. Do đó, ta được đường thẳng vì tất cả điểm nằm trên một đường thẳng.