基于条件的数组零填充

Question

我有两个数组：

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

我正在寻找的是根据以下条件填充零：

If the label[i-1] != label[i]:
   pad several zeros (say, 3) to the 'a' array at the same 'i' location

所以，我想要的结果是：

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

如您所见，数组 a 现在在值 7 之后有 3 个零，这些零由标签值已更改的条件填充。

我试过以下代码：

for i in range(len(a)):
    if label[i-1] != label[i]:         
        a = numpy.pad(a, (0,3), 'constant')
    else:
       pass

但是，零被填充在 a 数组的末尾。正如我怀疑的那样，我应该将填充操作等同于同一个数组，因为它在 for 循环中发生变化。

Answer 1

您需要在 label 的基础上进行更改，因此您需要遍历 label，而不是 a
现在你应该在 if 中添加 i != 0 条件，否则如果第一个和最后一个相同，你也会因为第一个元素而受到惩罚，因为 -1 返回到最后一个元素。

import numpy as np
a = np.array([ 1,  2,  3,  4,  5,  6,  7,  8,  9, 10])
label = np.array(['a', 'a', 'a', 'a', 'a', 'a', 'a', 'b', 'b', 'b'])

offset = 0
for i in range(len(label)):
    if i != 0 and label[i-1] != label[i]:
        len_ = 2 # no of 0's you want to add
        a = np.insert(a, i + offset, np.array([0] * len_))
        offset += len_
print(a)

输出：

[ 1  2  3  4  5  6  7  0  0  8  9 10]

Answer 2

这是一个基于 numpy 的方法：

def pad_at_diff(x, y, n):   
    # boolean mask where diffs occur 
    m = np.r_[False, y[:-1]!= y[1:]]
    # output array, expanded taking into account 
    # zeros to add
    x_pad = np.zeros(len(x)+n*len(m[m]))
    # assign at according indices adding cumsum of m
    x_pad[np.arange(len(x))+np.cumsum(m)*n] = x
    return x_pad

a = np.array([ 1,  2,  3,  4,  5,  6,  7,  8,  9, 10])
label = np.array(['a', 'a', 'a', 'a', 'a', 'a', 'a', 'b', 'b', 'b'])
pad_at_diff(a, label, 3)
array([ 1.,  2.,  3.,  4.,  5.,  6.,  7.,  0.,  0.,  0.,  8.,  9., 10.])

或者另一个例子：

a = np.array([ 1,  2,  3,  4,  5,  6,  7,  8,  9, 10,11,12])
label = np.array(['a', 'a', 'a', 'a', 'a', 'a', 'a', 'b', 'b', 'b', 'c', 'c'])
pad_at_diff(a, label, 3)
array([ 1.,  2.,  3.,  4.,  5.,  6.,  7.,  0.,  0.,  0.,  8.,  9., 10.,
        0.,  0.,  0., 11., 12.])

Answer 3

这是你想要的吗？

>>> for i in range(a.size-1):
        if label[i]!=label[i+1]:
            np.insert(a,i+1,[0]*3)

这是我得到的：

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

我参考了你的if条件

Answer 4

np 的 pad 函数添加到数组的末尾。我认为您正在寻找的是插入。插入的问题在于，一旦您插入值，您的索引就会随着循环的变化而变化。如果你从后面循环虽然它有效：

import numpy as np
a = np.array([ 1,  2,  3,  4,  5,  6,  7,  8,  9, 10])
label = np.array(['a', 'a', 'a', 'a', 'a', 'a', 'a', 'b', 'b', 'b'])

prev=None
for i,ele in enumerate(label[::-1]):
   if prev:
      if ele!=prev:
         a=np.insert(a,-i, [0,0,0])
         print(ele)
   prev=ele

基于条件的数组零填充

Condition based array zero padding

python

numpy

zero-padding