Question

我有一个关于python索引的问题：我正在尝试使用中心差分来估计＆＃39; dU＆＃39;来自一个阵列＆＃39; U＆＃39;我通过初始化＆＃39; dU＆＃39;来做到这一点。一系列的＆＃39; nan＆＃39;长度（U）然后应用中心差分使得dU（i）=（U（i + 1）-U（i-1））/ 2到中心元素。输出＆＃39; dU＆＃39;阵列目前给我两个＆＃39; nan＆＃39;向量末尾的条目。谁能解释为什么倒数第二个元素没有被更新？

import numpy as np
U= np.array([1,2,3,4,5,6])
dU = np.zeros(len(U))
dU[:] = np.NAN
dU[1:-2] = (U[2:-1]-U[0:-3])/2

>>> dU
array([ nan,   1.,   1.,   1.,  nan,  nan])

Answer 1

要包含倒数第二个元素，您需要：

dU[1:-1] = (U[2:]-U[0:-2])/2

Answer 2

没有回答您的问题，但作为一个有用的提示，您可以使用numpy.gradient

>>> np.gradient(np.array([1,2,3,4,5,6]))
>>> array([ 1.,  1.,  1.,  1.,  1.,  1.])

用于中心差分的Python索引

2 个答案: