2

我有以下计算矩阵的差异,即第 i 个元素 - (i-1) 元素。

如何(轻松)计算每个元素的水平和垂直差异?用转置?

inputarr = np.arange(12)
inputarr.shape = (3,4)
inputarr+=1

#shift one position
newarr = list()
for x in inputarr:
    newarr.append(np.hstack((np.array([0]),x[:-1])))

z = np.array(newarr)    
print inputarr
print 'first differences'
print inputarr-z

输出

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

first differences
[[1 1 1 1]
 [5 1 1 1]
 [9 1 1 1]]
4

1 回答 1

3

退房numpy.diff

从文档中:

计算沿给定轴的 n 阶离散差。

一阶差分out[n] = a[n+1] - a[n]沿给定轴给出,高阶差分通过递归使用 diff 计算。

一个例子:

>>> import numpy as np
>>> a = np.arange(12).reshape((3,4))
>>> a
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>> np.diff(a,axis = 1) # row-wise
array([[1, 1, 1],
       [1, 1, 1],
       [1, 1, 1]])
>>> np.diff(a, axis = 0) # column-wise
array([[4, 4, 4, 4],
       [4, 4, 4, 4]])
于 2012-12-22T15:57:20.013 回答