可视化python中两列之间的相关性答案

【问题标题】：Visualize correlation between two columns in python可视化python中两列之间的相关性
【发布时间】：2020-02-08 13:21:33
【问题描述】：

我有两列。第一列包含与薪水相关的数据，第二列包含与员工房屋租金相关的数据。现在使用 python 我想找到两者之间的相关性。 Python中是否有某种方法可以可视化相关性：

Salary   house_rent
10000    50
10000    50
3000     465

这项任务的重点是找出员工的工资和房租之间的相关性。例如。有的员工工资高但房租小，有的员工工资低但房租大。请注意，很可能两个人的工资和房租相同。是否可以在 python 中可视化？

【问题讨论】：

标签： python python-3.x matplotlib machine-learning statistics

【解决方案1】：

您可以使用 sklearn.linear_model.LinearRegression 绘制线性回归线：

https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LinearRegression.html

您还可以使用 matplotlib 中的 pyplot.matshow() 构建相关矩阵：

import matplotlib.pyplot as plt

plt.matshow(dataframe.corr())
plt.show()

【讨论】：

这里的数据框是什么？
dataframe 将是上述数据集的一个实例。只需将您的数据集存储为 csv 文件（例如使用 excel）并使用 pandas 库将其导入。将熊猫导入为 pd dataframe = pd.read_csv('file_name')

【解决方案2】：

如前所述，您可以使用pandas 中的corr 方法来获取相关性。更好的可视化方法是使用 seaborn 库而不是 matplotlib。

import seaborn as sns
import pandas as pd
import matplotlib.pyplot as plt
sns.set(style="ticks", color_codes=True)    
df= pd.read_csv('path_to_your_csv_file')
g = sns.pairplot(df)
plt.show()

【讨论】：

您需要致电plt.show()。我已经更新了答案。
它显示了 4 个图，每个图的含义是什么
它正在将所有列相互配对。请参阅答案中给出的链接或更好地在互联网上搜索配对图解释。