3

我在低通插值过程中遇到问题，我必须在处理之前用零填充一维数组。

我有这样的事情：[1 2 3 4 5 6]

我想要这样的数组 [1 0 2 0 3 0 4 0 5 0 6] 所以它是数组中的 L-1 个零，其中 L 是填充零之前数组中所有值的数量。

如何在 Python 中做到这一点？

4

6 回答 6

13

您可以将未填充的值列表分配到另一个零列表的切片中：

original_list = range(1,7)                 # [1,2,3,4,5,6]
padded_list = [0]*(2*len(original_list)-1) # [0,0,0,0,0,0,0,0,0,0,0]
padded_list[::2] = original_list           # [1,0,2,0,3,0,4,0,5,0,6]

这也可以转化为numpy，尽管正如 Jaime 在评论中指出的那样，它更容易使用numpy.insert：

import numpy as np

arr = np.arange(1, 7)              # array([1, 2, 3, 4, 5, 6])
np.insert(arr, slice(1, None), 0)  # array([1, 0, 2, 0, 3, 0, 4, 0, 5, 0, 6])
于 2013-04-24T09:58:39.743 回答
2

我必须实时应用大量数据，所以我检查了性能，np.insert 似乎是 Python 2.7 上最慢的一个：

np.insert:                     1.393 ms
list:                          0.406 ms
np.repeat then zero out:       0.409 ms
np.zeros then assign value:    0.073 ms

这是测试代码：

import time
import numpy as np

start = 0
limit = 30000000

def tic():
    global start
    start = time.clock()

def toc(text):
    print("%-30s %.3f ms" % (text, time.clock()-start))

original_list = range(1,limit)
original_array = np.arange(1, limit)

tic()
insert_array = np.insert(original_array, slice(1, None), 0)
toc("np.insert:")

tic()
padded_list = [0]*(2*len(original_list)-1)
padded_list[::2] = original_list
toc("list:")

tic()
repeat_array = np.repeat(original_array, 2)
repeat_array[1::2] = 0
toc("np.repeat then zero out:")

tic()
padded_array = np.zeros(2*len(original_list)-1, dtype=original_array.dtype)
padded_array[::2] = original_array
toc("np.zeros then assign value:")
于 2016-11-29T17:19:31.410 回答
0 
def zero(array):
    for i in range(1, len(array)):
        array.insert(i+i-1,0)
    return array

array = [1, 2, 3, 4, 5, 6]
newArray = zero(array)
于 2013-04-24T10:10:10.293 回答
0

强制itertools方法：

>>> from itertools import izip, repeat, chain
>>> list(chain.from_iterable(izip(el, repeat(0))))[:-1]
[1, 0, 2, 0, 3, 0, 4, 0, 5, 0, 6]

更好itertools的方法...

from itertools import islice, izip, repeat, chain
blah = list (islice (chain.from_iterable (izip(repeat (0), your_seq)), 1, None))
于 2013-04-25T23:55:05.060 回答
0 
l = [1, 2, 3, 4, 5, 6]

def zero_pad(l):
    r = []
    for x in l:
        r.append(x)
        r.append(0)
    return r[:-1]

print zero_pad(l)

将打印：

[1, 0, 2, 0, 3, 0, 4, 0, 5, 0, 6]
于 2013-04-24T09:57:10.570 回答
0

一种简单的方法：

>>> i = [1,2,3,4,5,6]
>>> [i.insert(i.index(x)+1,0) for x in i[:-1]]
[None, None, None, None, None]
>>> i
[1, 0, 2, 0, 3, 0, 4, 0, 5, 0, 6]

同上，展开：

>>> for x in i[:-1]:
...     i.insert(i.index(x)+1,0)
...
>>> i
[1, 0, 2, 0, 3, 0, 4, 0, 5, 0, 6]
于 2013-04-24T09:58:37.443 回答