12个最常用的matplotlib图例 ！！
1、折线图
折线图（Line Plot）：用于显示数据随时间或其他连续变量的变化趋势。在实际项目中，可以用于可视化模型性能随着训练迭代次数的变化。
下面的示例中，我们将绘制一个包含多个数据系列的折线图。
首先，确保已经安装了Matplotlib库。（pip install matplotlib）
import matplotlib.pyplot as plt
import numpy as np
# 创建示例数据集
x = np.linspace(0, 10, 100) # 生成0到10之间的100个数据点作为x轴
y1 = np.sin(x) # 第一个数据系列，正弦函数
y2 = np.cos(x) # 第二个数据系列，余弦函数
y3 = np.sin(2 * x) # 第三个数据系列，正弦函数的倍频
# 创建一个Matplotlib图表
plt.figure(figsize=(10, 6)) # 设置图表的大小
# 绘制折线图
plt.plot(x, y1, label='sin(x)', color='blue', linestyle='-', linewidth=2)
plt.plot(x, y2, label='cos(x)', color='red', linestyle='--', linewidth=2)
plt.plot(x, y3, label='sin(2x)', color='green', linestyle='-.', linewidth=2)
# 添加标题和标签
plt.title('Complex Line Plot')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
# 添加图例
plt.legend()
# 自定义坐标轴范围
plt.xlim(0, 10)
plt.ylim(-2, 2)
# 添加网格线
plt.grid(True, linestyle='--', alpha=0.6)
# 保存图像（可选）
# plt.savefig('complex_line_plot.png')
# 显示图像
plt.show()包含三个不同的数据系列，每个系列都有不同的线型和颜色。
（可以根据需要自定义图表的样式、颜色和标签）
2、散点图
散点图（Scatter Plot）：用于显示两个变量之间的关系，通常用于观察数据的分布、异常值或类别之间的关系。
创建一个或多个包含数据系列的散点图。
我们将使用随机生成的数据集来模拟复杂的散点图：
import matplotlib.pyplot as plt
import numpy as np
# 创建示例数据集
np.random.seed(0)
n_points = 200
x = np.random.rand(n_points) # 随机生成x坐标
y1 = np.random.rand(n_points) # 随机生成y坐标
y2 = 1.5 * x + np.random.rand(n_points) # 生成带趋势的数据
colors = np.random.rand(n_points) # 随机生成颜色值
sizes = np.random.rand(n_points) * 100 # 随机生成点的大小
# 创建一个Matplotlib图表
plt.figure(figsize=(10, 6)) # 设置图表的大小
# 绘制散点图
plt.scatter(x, y1, label='Random Data', c=colors, s=sizes, alpha=0.7, cmap='viridis')
plt.scatter(x, y2, label='Trendline Data', c='red', marker='x', s=50, alpha=0.7)
# 添加标题和标签
plt.title('Complex Scatter Plot')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
# 添加图例
plt.legend()
# 自定义坐标轴范围
plt.xlim(0, 1)
plt.ylim(0, 2)
# 添加颜色条
colorbar = plt.colorbar()
colorbar.set_label('Color Intensity')
# 保存图像（可选）
# plt.savefig('complex_scatter_plot.png')
# 显示图像
plt.show()上面代码创建了一个复杂的散点图，其中包含两个不同的数据系列，每个系列都具有不同的颜色、标记和大小。通过使用cmap函数，可以改变数据点的颜色映射。
3、直方图
直方图（Histogram）：用于显示数据的分布情况，特别适用于展示数值型数据的频率分布。
使用多个数据系列以及自定义的颜色、透明度和边界线等属性来创建一个直方图。
import matplotlib.pyplot as plt
import numpy as np
# 创建示例数据集（多个数据系列）
np.random.seed(0)
data_series = [np.random.normal(0, std, 1000) for std in range(1, 4)]
# 创建一个Matplotlib图表
plt.figure(figsize=(10, 6)) # 设置图表的大小
# 绘制直方图
plt.hist(data_series, bins=20, color=['blue', 'green', 'purple'], alpha=0.7, label=['Series 1', 'Series 2', 'Series 3'], edgecolor='black')
# 添加标题和标签
plt.title('Complex Histogram')
plt.xlabel('Value')
plt.ylabel('Frequency')
# 添加图例
plt.legend()
# 自定义坐标轴范围
plt.xlim(-4, 4)
plt.ylim(0, 300)
# 添加网格线
plt.grid(True, linestyle='--', alpha=0.6)
# 保存图像（可选）
# plt.savefig('complex_histogram.png')
# 显示图像
plt.show()上述代码中，包含三个不同的数据系列，每个系列都具有不同的颜色、透明度和边界线颜色。直方图的bins数设置为20，可以根据需要进行调整。
4、柱状图
柱状图（Bar Plot）：用于比较不同类别之间的数据，例如不同产品的销售量或不同类别的统计i数据。
当涉及到柱状图可视化时，Matplotlib提供了丰富的自定义选项。
下面代码将创建一个具有多个数据系列、堆叠柱状图和自定义颜色、标签等属性的柱状图。
import matplotlib.pyplot as plt
import numpy as np
# 创建示例数据集（多个数据系列）
categories = ['Category A', 'Category B', 'Category C', 'Category D']
data_series1 = [3, 4, 2, 6]
data_series2 = [2, 5, 1, 4]
data_series3 = [1, 3, 4, 2]
# 创建一个Matplotlib图表
plt.figure(figsize=(10, 6)) # 设置图表的大小
# 绘制柱状图（堆叠）
bar_width = 0.2
index = np.arange(len(categories))
plt.bar(index, data_series1, bar_width, label='Series 1', color='blue', edgecolor='black')
plt.bar(index, data_series2, bar_width, label='Series 2', color='green', edgecolor='black', bottom=data_series1)
plt.bar(index, data_series3, bar_width, label='Series 3', color='purple', edgecolor='black', bottom=np.array(data_series1) + np.array(data_series2))
# 添加标题和标签
plt.title('Complex Bar Chart')
plt.xlabel('Categories')
plt.ylabel('Values')
# 添加图例
plt.legend()
# 自定义坐标轴标签
plt.xticks(index, categories)
# 自定义坐标轴范围
plt.ylim(0, 15)
# 添加网格线
plt.grid(True, linestyle='--', alpha=0.6)
# 保存图像（可选）
# plt.savefig('complex_bar_chart.png')
# 显示图像
plt.show()示例中，使用了三个数据系列，每个系列都具有不同的颜色。柱状图被堆叠在一起，以显示每个类别中各系列的值，并使用bottom参数来堆叠。
5、箱线图
箱线图（Box Plot）：用于展示数据的分布、中位数、离群值等统计信息，有助于检测数据中的异常值。
import matplotlib.pyplot as plt
import numpy as np
# 创建一个示例数据集，这里使用随机生成的数据
np.random.seed(0)
data = [np.random.normal(0, std, 100) for std in range(1, 4)]
# 设置箱线图的标签
labels = ['Dataset 1', 'Dataset 2', 'Dataset 3']
# 创建一个箱线图
fig, ax = plt.subplots(figsize=(10, 6))
# 自定义箱线图的外观，包括颜色、填充、标记和线宽
boxprops = dict(linewidth=2, color='blue', facecolor='lightblue')
flierprops = dict(marker='o', markerfacecolor='red', markersize=8, linestyle='none')
medianprops = dict(linestyle='-', linewidth=2, color='green')
whiskerprops = dict(color='blue', linewidth=2)
capprops = dict(color='blue', linewidth=2)
# 画出箱线图
bplot = ax.boxplot(data, vert=True, patch_artist=True, labels=labels,
boxprops=boxprops, flierprops=flierprops,
medianprops=medianprops, whiskerprops=whiskerprops,
capprops=capprops)
# 自定义Y轴标签
ax.set_ylabel('Values')
# 添加标题
ax.set_title('Complex Box Plot')
# 添加网格线
ax.yaxis.grid(True, linestyle='--', which='major', color='gray', alpha=0.7)
# 自定义箱线图的背景色
colors = ['lightblue', 'lightgreen', 'lightpink']
for patch, color in zip(bplot['boxes'], colors):
patch.set_facecolor(color)
# 自定义X轴刻度标签
ax.set_xticklabels(labels)
# 显示图例
legend_elements = [plt.Line2D([0], [0], marker='o', color='w', markerfacecolor='red', markersize=10, label='Outliers')]
ax.legend(handles=legend_elements, loc='upper right')
# 自定义X轴标签位置
ax.set_xticks(np.arange(1, len(labels) + 1))
# 调整Y轴刻度范围，根据的数据集进行调整
ax.set_ylim(-5, 10)
# 显示图形
plt.tight_layout()
plt.show()上面代码创建不同颜色的箱体、自定义标记的异常值、自定义线条属性和其他个性化选项。
6、热力图
热力图（Heatmap）：用于可视化矩阵数据，通常用于显示相关性、相似矩阵或特征之间的关系。
import matplotlib.pyplot as plt
import numpy as np
# 创建示例数据集（这里使用随机生成的数据）
np.random.seed(0)
data = np.random.rand(10, 10)
# 设置热力图的标签和自定义参数
labels = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J']
cmap = 'YlOrRd' # 自定义颜色映射
annot = True # 显示数值标签
fmt = '.2f' # 数值标签格式
# 创建热力图
fig, ax = plt.subplots(figsize=(8, 8))
# 自定义热力图的外观
heatmap = ax.imshow(data, cmap=cmap, aspect='auto')
# 显示数值标签
if annot:
for i in range(len(labels)):
for j in range(len(labels)):
text = ax.text(j, i, format(data[i, j], fmt),
ha="center", va="center", color="black")
# 添加颜色条
cbar = plt.colorbar(heatmap)
# 自定义X轴和Y轴标签
ax.set_xticks(np.arange(len(labels)))
ax.set_yticks(np.arange(len(labels)))
ax.set_xticklabels(labels)
ax.set_yticklabels(labels)
# 添加标题
ax.set_title('Customized Heatmap')
# 自定义热力图的颜色条标签
cbar.set_label('Values', rotation=270, labelpad=15)
# 显示图形
plt.tight_layout()
plt.show()上面代码中提供了多个定制化参数，包括颜色映射、数值标签、数值标签格式、颜色条标签等。可以根据自己的数据集和需求进一步自定义热力图，例如更改颜色映射、调整数值标签格式、添加自定义标题等。
7、饼图
饼图（Pie Chart）：用于显示数据的部分与整体的比例，通常用于显示类别的占比。
import matplotlib.pyplot as plt
# 创建示例数据集（这里使用随机生成的数据）
labels = ['Category A', 'Category B', 'Category C', 'Category D']
sizes = [15, 30, 45, 10] # 自定义每个类别的大小
explode = (0.1, 0, 0, 0) # 自定义饼图中的分离
# 自定义颜色，可以使用Matplotlib中的其他颜色映射
colors = ['#ff9999', '#66b3ff', '#99ff99', '#c2c2f0']
# 自定义百分比格式
autopct = '%1.1f%%'
# 创建饼图
fig, ax = plt.subplots()
# 自定义饼图的外观
ax.pie(sizes, explode=explode, labels=labels, colors=colors, autopct=autopct,
shadow=True, startangle=90, wedgeprops={'edgecolor': 'gray'})
# 添加标题
ax.set_title('Customized Pie Chart')
# 显示图形
plt.tight_layout()
plt.show()上述的代码中，提供了多个定制化参数，包括每个类别的大小、分离、颜色、百分比格式等。可以根据自己的数据集和需求进一步自定义饼图，例如更改颜色、调整百分比格式、添加自定义标题等。
8、面积图
面积图（Area Plot）:类似于折线图，但可以用于展示多个类别之间的堆积关系，通常用于实践序列数据。
import matplotlib.pyplot as plt
import numpy as np
# 创建示例数据集
years = [2010, 2011, 2012, 2013, 2014]
categories = ['Category A', 'Category B', 'Category C', 'Category D']
data = np.array([
[5, 8, 6, 9, 12],
[7, 10, 8, 11, 14],
[4, 6, 5, 7, 9],
[3, 5, 4, 6, 7]
])
# 自定义堆积颜色
colors = ['#ff9999', '#66b3ff', '#99ff99', '#c2c2f0']
# 创建面积堆积图
fig, ax = plt.subplots(figsize=(10, 6))
# 绘制面积堆积图
bottoms = np.zeros(len(years))
for i, category in enumerate(categories):
ax.fill_between(years, bottoms, bottoms + data[i], label=category, color=colors[i], alpha=0.7)
bottoms += data[i]
# 添加标题
ax.set_title('Customized Stacked Area Chart (Complex Data)')
# 自定义X轴和Y轴标签
ax.set_xlabel('Year')
ax.set_ylabel('Values')
# 添加图例
ax.legend(loc='upper left')
# 添加阴影
ax.grid(alpha=0.3)
# 自定义Y轴刻度
ax.set_yticks(np.arange(0, max(bottoms), step=10))
# 自定义X轴刻度标签
ax.set_xticks(years)
# 显示图形
plt.tight_layout()
plt.show()上述代码中，包括数据集、自定义颜色、图例、标签、标题、阴影和刻度标签等。
9、等高线图
等高线图（Contour Plot）：用于可视化二维数据的等高线，通常用于显示函数的等值线。
import matplotlib.pyplot as plt
import numpy as np
# 创建示例数据集
x = np.linspace(-2, 2, 400)
y = np.linspace(-2, 2, 400)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2)) * np.exp(-0.2 * np.sqrt(X**2 + Y**2))
# 自定义等高线参数
levels = np.linspace(-1, 1, 20)
cmap = plt.get_cmap('RdYlBu_r') # 自定义颜色映射
# 创建等高线图
fig, ax = plt.subplots(figsize=(8, 8))
# 绘制等高线图
contour = ax.contourf(X, Y, Z, levels=levels, cmap=cmap)
# 添加等高线线条
contour_lines = ax.contour(X, Y, Z, levels=levels, colors='k', linewidths=0.5)
# 添加颜色条
cbar = plt.colorbar(contour, ax=ax, ticks=np.arange(-1, 1.2, 0.2))
cbar.set_label('Values')
# 添加标题
ax.set_title('Customized Contour Plot')
# 自定义X轴和Y轴标签
ax.set_xlabel('X-axis')
ax.set_ylabel('Y-axis')
# 添加网格线
ax.grid(linestyle='--', alpha=0.7)
# 添加等高线标签
ax.clabel(contour_lines, inline=1, fontsize=8, fmt='%0.1f')
# 显示图形
plt.tight_layout()
plt.show()上述代码中创建了一个等高线图，包括自定义等高线参数、颜色映射、颜色条、等高线线条、标签、标题和网格线等。
10、3D图
3D图（3D Plot）：用于可视化包含三个变量的数据，例如三维散点图、三维曲面图等。
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
# 创建示例数据集
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2)) / (np.sqrt(X**2 + Y**2) + 1)
# 创建3D图形
fig = plt.figure(figsize=(10, 8))
ax = fig.add_subplot(111, projection='3d')
# 自定义颜色映射和标签
cmap = plt.get_cmap('viridis')
ax.plot_surface(X, Y, Z, cmap=cmap, label='Surface Plot')
# 添加颜色条
cbar = fig.colorbar(ax.plot_surface(X, Y, Z, cmap=cmap), ax=ax, pad=0.1)
cbar.set_label('Values')
# 添加标题
ax.set_title('Customized 3D Surface Plot')
# 自定义坐标轴标签
ax.set_xlabel('X-axis')
ax.set_ylabel('Y-axis')
ax.set_zlabel('Z-axis')
# 自定义坐标轴刻度
ax.set_xticks(np.arange(-5, 6, 2))
ax.set_yticks(np.arange(-5, 6, 2))
# 自定义视角
ax.view_init(elev=20, azim=45) # 仰角和方位角
# 显示图形
plt.tight_layout()
plt.show()上述代码中，创建了一个复杂的3D表面绘图，包括自定义颜色映射、颜色条、标签、标题、坐标轴标签、坐标轴刻度和视角。
11、时间序列图
时间序列图（Time Series Plot）：用于可视化时间序列数据，通常包括折线图和柱状图，以便观察时间趋势和季节性模式。
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
# 创建示例数据集
np.random.seed(0)
dates = pd.date_range('2023-01-01', '2023-12-31', freq='D')
num_series = 5
data = np.random.randn(len(dates), num_series).cumsum(axis=0)
# 创建时间序列图
fig, ax = plt.subplots(figsize=(12, 6))
# 自定义线条颜色和样式
colors = ['b', 'g', 'r', 'c', 'm']
linestyles = ['-', '--', '-.', ':', '-']
for i in range(num_series):
ax.plot(dates, data[:, i], label=f'Series {i+1}', color=colors[i], linestyle=linestyles[i])
# 添加标题
ax.set_title('Customized Time Series Plot')
# 自定义X轴和Y轴标签
ax.set_xlabel('Date')
ax.set_ylabel('Value')
# 添加图例
ax.legend(loc='upper left')
# 添加网格线
ax.grid(linestyle='--', alpha=0.7)
# 自定义X轴的日期刻度显示
ax.xaxis.set_major_locator(plt.MaxNLocator(10)) # 最多显示10个日期刻度
# 自定义日期刻度标签的格式
from matplotlib.dates import DateFormatter
date_format = DateFormatter('%b %d')
ax.xaxis.set_major_formatter(date_format)
# 自定义Y轴刻度范围
ax.set_ylim(-10, 10)
# 显示图形
plt.tight_layout()
plt.show()上述代码中，自定义线条颜色和样式、标签、标题、坐标轴标签、图例、网格线、日期刻度显示和日期刻度标签的格式。
12、树状图
树状图（Tree Disgram）：用于可视化决策树、层次聚类等树状结构的数据。
使用 networkx 库来创建树的结构，并使用 matplotlib 进行可视化 。 （pip install networkx）
import matplotlib.pyplot as plt
import networkx as nx
# 创建示例数据集
G = nx.DiGraph()
G.add_nodes_from([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
G.add_edges_from([(1, 2), (1, 3), (2, 4), (2, 5), (3, 6), (3, 7), (4, 8), (4, 9), (5, 10)])
# 创建树状图布局
pos = nx.spring_layout(G, seed=42)
# 自定义节点颜色和大小
node_colors = ['#1f78b4' for _ in G.nodes()]
node_sizes = [800 for _ in G.nodes()]
# 创建树状图
fig, ax = plt.subplots(figsize=(10, 8))
# 绘制树状图
nx.draw(G, pos, ax=ax, with_labels=True, node_color=node_colors, node_size=node_sizes,
font_size=10, font_weight='bold', edge_color='gray', width=1.0, arrows=True)
# 添加标题
ax.set_title('Customized Tree Diagram')
# 显示图形
plt.tight_layout()
plt.show()上述代码中，包括自定义数据集、自定义节点颜色、大小、标签、标题、布局、边的颜色和宽度等。
总结
这些图表类型覆盖了数据分析和机器学习项目的许多常见需求。根据具体的项目和数据，可以选择适合的图表类型来展示和分析数据。Matplotlib提供 了丰富的功能，能够自定义图表以满足特定的需求。
参考：深夜努力写Python
本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2024-02-09，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读