博客

numpy中pad函数的常用方法

  • Post category:other

NumPy中的pad函数可以用于在数组的边缘添加值。它可以用于填充缺失的数据、扩展数组的大小或为卷积操作准备数据。本文将介绍NumPy中pad函数的常用方法，包括填充模式、填充值和填充维度等。

填充模式

pad函数的第二个参数指定了填充模式。以下是常用的填充模式：

  • ‘constant’：用常数填充数组的边缘。
  • ‘edge’：用数组的边缘值填充数组的边缘。
  • ‘linear_ramp’：用线性渐变填充数组的边缘。
  • ‘maximum’：用数组的最大值填充数组的边缘。
  • ‘mean’：用数组的平均值填充数组边缘。
  • ‘median’：用数组的中位数填充数组的边缘。
  • ‘minimum’：用数组的最小值填充数组的边缘。
  • ‘reflect’：用数组的反射值填充数组的边缘。
  • ‘symmetric’：用数组的对称值填充数组的边缘。
  • ‘wrap’：用数组的循环值填充数组的边缘。

以下是一个示例，演示如何使用’constant’模式填充数组的边缘：

import numpy as np

arr = np.array([[1, 2], [3, 4]])
padded_arr = np.pad(arr, ((1, 1), (1, 1)), mode='constant', constant_values=0)

print(padded_arr)

输出：

array([[0, 0, 0, 0],
       [0, 1, 2, 0],
       [0, 3, 4, 0],
       [0, 0, 0, 0]])

在上面的示例中，我们使用’constant’模式填了一个2×2的数组。我们将填充宽度设置为1，并将填充值设置为0。

填充值

pad函数的第三个参数指定了填值。以下是一些常用的填充值：

  • 数字：用数字填充数组的边缘。
  • 序列：用序列中的值填充数组的边缘。
  • 字符串：用字符串填充数组的边缘。

以下是一个示例，演示如何使用数字填充数组的边缘：

import numpy as np

arr = np.array([[1, 2], [3, 4]])
padded_arr = np.pad(arr, ((1, 1), (1, 1)), mode='constant', constant_values=0)

print(padded_arr)

输出：

array([[0, 0, 0, 0],
       [0, 1, 2, 0],
       [0, 3, 4, 0],
       [0, 0, 0, 0]])

在上面的示例中，我们使用数字0填充了一个2×2的数组。我们将填充宽度设置为1，并将填充值设置为0。

填充维度

pad函数的第一个参数指定了要填充的数组。它可以是一维、二维或多维数组。以下是一个示例，演示如何使用pad函数填充一维数组：

import numpy as np

arr = np.array([1, 2, 3])
padded_arr = np.pad(arr, (2, 3), mode='constant', constant_values=0)

print(padded_arr)

输出：

array([0, 0, 1, 2, 3, 0, 0, 0])

在上面的示例中，我们使用函数填充了一个一维数组。我们将填充宽度设置为2和3，并将填充值设置为0。

示例一

以下是一个示例，演示如何使用’edge’模式填充数组的边缘：

import numpy as np

arr = np.array([[1, 2], [3, 4]])
padded_arr = np.pad(arr, ((1, 1), (1, 1)), mode='edge')

print(padded_arr)

输出：

array([[1, 1, 2, 2],
       [1, 1, 2, 2],
       [3, 3, 4, 4],
       [3, 3, 4, 4]])

在上面的示例中，我们使用’edge’模式填充了一个2×2的数组。我们将填充宽度设置为1。

示例二

以下是一个示例，演示如何使用’wrap’模式填充数组的边缘：

import numpy as np

arr = np.array([[1, 2], [3, 4]])
padded_arr = np.pad(arr, ((1, 1), (1, 1)), mode='wrap')

print(padded_arr)

输出：

array([[4, 3, 4, 3],
       [2, 1, 2, 1],
       [4, 3, 4, 3],
       [2, 1, 2, 1]])

在上面的示例中，我们使用’wrap’模式填充了一个2×2的数组。我们将填充宽度设置为1。

总结

NumPy中的pad函数可以用于在数组的边缘添加值。它可以用于填充缺失的数据、扩展数组的大小或为卷积操作准备数据。在使用pad函数时，需要指定填充模式、填充值和填充维度等参数。常用的填充模式包括’constant’、’edge’、’linear_ramp’、’maximum’、’mean’、’median’、’minimum’、’reflect’、’symmetric’和’wrap’。填充值可以是数字、序列或字符串。填充维度可以是一维、二维或多维数组。

Tags: