python:for循环紧凑表示

Question

Python, NumPy

有没有更紧凑的方法来操作数组元素，而不必使用标准的 for 循环？

例如，考虑下面的函数:

filterData(A):
    B = numpy.zeros(len(A));
    B[0] = (A[0] + A[1])/2.0;
    for i in range(1, len(A)): 
        B[i] = (A[i]-A[i-1])/2.0;
    return B;

Answer 1

Numpy 有一个 diff operator适用于 numpy 数组和 Python native 数组。您可以将代码重写为:

def filterData(A):
    B = numpy.zeros(len(A));
    B[1:] = np.diff( A )/2.0
    B[0] = (A[0] + A[1])/2.0;
    return B