当前位置: 首页 > 面试题库 >

Python-从numpy数组中获取具有给定步幅/步长的子数组

松旭
2023-03-14
问题内容

可以说我有一个Python Numpy数组a

a = numpy.array([1,2,3,4,5,6,7,8,9,10,11])

我想从步长为3的长度为5的数组创建一个子序列矩阵,因此结果矩阵如下所示:

numpy.array([[1,2,3,4,5],[4,5,6,7,8],[7,8,9,10,11]])

实现此目的的一种可能方法是使用faor循环。

result_matrix = np.zeros((3, 5))
for i in range(0, len(a), 3):
  result_matrix[i] = a[i:i+5]

有没有更干净的方法可以在Numpy中实现呢


问题答案:

方法#1: 使用broadcasting

def broadcasting_app(a, L, S ):  # Window len = L, Stride len/stepsize = S
    nrows = ((a.size-L)//S)+1
    return a[S*np.arange(nrows)[:,None] + np.arange(L)]

方法2:使用更有效的方法NumPy strides

def strided_app(a, L, S ):  # Window len = L, Stride len/stepsize = S
    nrows = ((a.size-L)//S)+1
    n = a.strides[0]
    return np.lib.stride_tricks.as_strided(a, shape=(nrows,L), strides=(S*n,n))

样品运行

In [143]: a
Out[143]: array([ 1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11])

In [144]: broadcasting_app(a, L = 5, S = 3)
Out[144]: 
array([[ 1,  2,  3,  4,  5],
       [ 4,  5,  6,  7,  8],
       [ 7,  8,  9, 10, 11]])

In [145]: strided_app(a, L = 5, S = 3)
Out[145]: 
array([[ 1,  2,  3,  4,  5],
       [ 4,  5,  6,  7,  8],
       [ 7,  8,  9, 10, 11]])


 类似资料:
  • 我有一个数组,并希望将其广播到,但仅适用于的情况。否则应该用零填充。当然,我可以创建新的空数组并用填充它,如下所示: 不幸的是,这两个方法都用值填充所有对。 我的问题是,如果这是可能的创建数组的形状我需要使用大步和没有实际复制数据?

  • 问题内容: 给定一个非负整数数组和一个数字。您需要打印总和等于给定整数的子数组的所有开始和结束索引。 例如 : Explanation : [3, 6] [9], [9,0] These all are the subarrays with their sum equal to 9. 问题答案: 解决这个问题的基本蛮力方法是生成给定数组的所有子数组,然后遍历生成的子数组并计算总和,如果这个总和等于

  • 问题内容: 我有一个二维的numpy数组,具有相等数量的列和行。我想将它们排列成一个较大的数组,对角线上的数组较小。应该可以指定起始矩阵在对角线上的频率。例如: 因此,如果我希望此数组在对角线上2次,则期望的输出将是: 3次: 有没有一种快速的方法来使用numpy方法以及对于任意大小的起始数组(仍然考虑到起始数组具有相同的行数和列数)来实现这一点? 问题答案: 方法1 经典案例- 样品运行- 方法

  • 问题内容: 给定一个大小为numpy的数组和一个整数,我想生成该数组的所有顺序长度子序列,最好是二维数组。 例: 我能想到的最好方法是 我是否缺少更好的(最好是内置的)功能? 问题答案: 这是一种非常快速且内存有效的方法,仅是对原始数组的“查看”: 如果需要写入此数组,则应先做一个步骤,否则,您将修改原始数组和“子序列”数组中的相应条目。 此处更多信息:https : //stackoverflo

  • 问题内容: 我正在使用Python(通过)包装的C库来运行一系列计算。在运行的不同阶段,我想将数据放入Python,尤其是数组。 我正在使用的包装对数组数据进行两种不同类型的返回(这对我特别有意义): Array:当我这样做时(其中x是数组,我得到一个回报。我知道该数据是文档中内部数据的副本,我能够轻松地将其放入数组中: 这将返回数据的一维数组。 指向数据的指针:在这种情况下,从库的文档中可以了解

  • 问题内容: 我有并想要一些函数,调用的结果是新数组。 问题答案: 看一眼