如何在matplotlib中调整图形大小|极客教程

如何在matplotlib中调整图形大小

参考：how to adjust figure size in matplotlib

在使用matplotlib进行数据可视化时，我们经常需要调整绘图的大小以适应不同的需求和展示效果。本文将详细介绍如何在matplotlib中调整图形的大小，包括调整整个图形的大小和调整子图的大小。

调整整个图形的大小

我们可以使用figure()函数来创建一个新的图形，并通过设置figsize参数来调整图形的大小。figsize参数需要传入一个元组，包含图形的宽度和高度，单位是英寸。

import matplotlib.pyplot as plt

# 创建一个大小为8x6英寸的新图形
plt.figure(figsize=(8, 6))

示例代码2

除了在创建图形时设置大小外，我们也可以在已有图形中调整大小。首先使用gcf()函数获取当前图形，然后通过set_size_inches()方法设置新的大小。

import matplotlib.pyplot as plt

# 创建一个默认大小的图形
plt.plot([1, 2, 3, 4])
plt.show()

# 获取当前图形并设置为10x8英寸
fig = plt.gcf()
fig.set_size_inches(10, 8)

Output:

如何在matplotlib中调整图形大小

示例代码3

在某些情况下，我们可能需要将图形保存为一个文件，这时也可以设置图形的大小。使用savefig()函数保存图形时，可以通过figsize参数来设置图片的大小。

import matplotlib.pyplot as plt

# 创建一个图形并保存为png文件，大小为12x10英寸
plt.plot([1, 2, 3, 4])
plt.savefig('my_plot.png', figsize=(12, 10))

示例代码4

在使用subplots()函数创建子图时，也可以通过figsize参数来设置整个图形的大小。

import matplotlib.pyplot as plt

# 创建一个包含两个子图的图形，大小为10x8英寸
fig, axs = plt.subplots(2, figsize=(10, 8))

调整子图的大小

当我们需要在同一个图形中展示多个子图时，可以使用subplot()函数或subplots()函数创建子图。我们可以通过设置gridspec_kw参数来调整子图的大小和排列。

import matplotlib.pyplot as plt

# 创建一个2x2的子图，第一个子图占据整个第一行
fig = plt.figure(figsize=(8, 6))
ax1 = fig.add_subplot(2, 2, 1)
ax1.set_position([0, 0, 0.5, 1])

ax2 = fig.add_subplot(2, 2, 2)
ax3 = fig.add_subplot(2, 2, 3)
ax4 = fig.add_subplot(2, 2, 4)

示例代码6

通过subplots()函数创建子图时，可以设置gridspec_kw参数来调整子图的大小和布局。在示例中，我们创建了2×2的子图，并将第一个子图放大到占据整个第一行。

import matplotlib.pyplot as plt

# 创建一个2x2的子图，第一个子图占据整个第一行
fig, axs = plt.subplots(2, 2, figsize=(8, 6), gridspec_kw={'width_ratios': [2, 1]})

示例代码7

在某些情况下，我们可能需要在同一个图形中展示不同大小的子图。使用add_axes()方法可以在指定位置添加子图，然后通过设置位置参数来调整子图的大小。

import matplotlib.pyplot as plt

# 创建一个包含两个不同大小子图的图形
fig = plt.figure(figsize=(8, 6))
ax1 = fig.add_axes([0.1, 0.1, 0.4, 0.4])
ax2 = fig.add_axes([0.6, 0.6, 0.3, 0.3])

示例代码8

我们还可以使用subplot2grid()函数创建一个网格布局，并通过设置colspan和rowspan参数来调整子图的大小。

import matplotlib.pyplot as plt

# 创建一个2x2的子图网格，第一行第一列子图占据第一行和第一列
ax1 = plt.subplot2grid((2, 2), (0, 0), colspan=2)
ax2 = plt.subplot2grid((2, 2), (1, 0))
ax3 = plt.subplot2grid((2, 2), (1, 1))