Menu
CoddyTech

回顾:冰淇淋与阳光

本课属于 Coddy Machine Learning 学习旅程的“机器学习统计学”单元。第 34 课,共 64 课。

challenge icon

挑战

中等

现在来看看每组关系构成的矩阵。海滩小亭记录了 40 天的数据,保存在 kiosk.csv 中:包含一个 date 列和六个数值列。程序会读取一个列名。

  1. 输出与它关系最密切的其他列,不考虑正负号。
  2. 按表格顺序检查每对列,输出关系最密切的一对不同列,每对只检查一次。
  3. 将矩阵绘制成从 -1 到 1 的 coolwarm 热力图,添加颜色条,将列名设为刻度标签,并将标题设为 Kiosk Correlations。

将两个 r 都四舍五入到小数点后 2 位。对于更小的数据记录,输入 temp_c 时:

Most related to temp_c: ice_creams (r = 0.87)
Strongest pair: visitors and ice_creams (r = 0.96)

图表如下(Coddy 接下来会介绍你绘制的图表):

一张 3 乘 3 的热力图,标题为 Kiosk Correlations,刻度标签为 temp_c、visitors 和 ice_creams,每个单元格都是红色的某种深浅,其中 visitors 和 ice_creams 对应的非对角线单元格颜色最深

亲手试试

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

kiosk = pd.read_csv("kiosk.csv")
column = input()

# 在下方编写你的代码

机器学习统计学中的所有课

自己练习: Python 在线编译器