python学习04之柱形图和热图

古城微笑少年丶 2021-11-29 15:16 171阅读 0赞

1、加载数据

和折线图一样，我们首先都要将csv文件中的数据加载出来（代码是来自kaggle微教程中的）

# Path of the file to read
    flight_filepath = "../input/flight_delays.csv"
    
    # Read the file into a variable flight_data
    flight_data = pd.read_csv(flight_filepath, index_col="Month")

你们可能会注意到，代码比我们在折线图中使用的代码略短。因为在本例中，由于行标签(来自“Month”列)不与日期对应，所以我们不在括号中添加parse\_dates=True。

但是，我们像以前一样保留前两段文本，以便同时提供这两段:

数据集的文件路径(在本例中为flight\_filepath)和将用于索引行的列的名称(在本例中，index\_col="Month")

2、检查数据

print(flight_data)

3、条形图

# 设置长度和宽度
    plt.figure(figsize=(10,6))
    
    # 添加条形图的标签
    plt.title("Average Arrival Delay for Spirit Airlines Flights, by Month")
    
    # 标注X轴和Y轴的数据名
    sns.barplot(x=flight_data.index, y=flight_data['NK'])
    
    # 添加垂直轴的标签
    plt.ylabel("Arrival delay (in minutes)")

第三行代码解析：

它有三个主要组成部分:

sns.barplot就是要告诉笔记本我们想要创建一个条形图。请记住，sns引用seaborn包，本课程中用于创建图表的所有命令都以这个前缀开始。

x = flight\_data.index—这决定在水平轴上使用什么。在本例中，我们选择了索引行的列(在本例中，是包含月份的列)。

y=flight\_data\['NK'\] -设置数据中的列，用于确定每个栏的高度。在本例中，我们选择“NK”列。

注意:必须选择索引列flight\_data。索引，并且不可能使用flight\_data\['Month'\](它将返回一个错误)。

这是因为当我们加载数据集时，“Month”列用于索引行。我们总是必须使用这种特殊的符号来选择索引列。

4、热图

#设置图形的宽度和高度
    plt.figure(figsize=(14,7))
    
    # 添加标签
    plt.title("Average Arrival Delay for Each Airline, by Month")
    
    # 热图显示了每个航空公司平均每个月的到达延迟
    sns.heatmap(data=flight_data, annot=True)
    
    # 添加水平轴的标签
    plt.xlabel("Airline")

第三行代码解析：

该代码有三个主要组件:

sns.heatmap就是告诉笔记本，我们想创建一个热图。

*　　　　　　　　　　　　*data=flight\_data——这告诉笔记本使用flight\_data中的所有条目来创建热图。  
　　　　　　　　　　　　annot=True——这确保每个单元格的值都出现在图表上。(省略这一项将删除每个单元格中的数字!)

转载于:https://www.cnblogs.com/fb1704011013/p/11178193.html