python numpy 列表操作

Question

我有以下 numpy 数组:

import numpy as np
a = np.array([2,15,66,180])

现在，我希望连续元素之间的差异 (diff) 除以 25，始终小于或等于 1 (diff/25 < 0)。

out = some_function(a,25)
out

[2,15,40,65,66,91,116,141,166,180]

有没有智能的 pythonic 方法来做到这一点？

Answer 1

Numpy arange函数可能在那里有用，我会按照以下方式进行:

import numpy as np

a = np.array([2,15,66,180])
out = np.array([])
for i in range(len(a)-1):
    out = np.hstack((out,np.arange(a[i],a[i+1],25)))
out = np.hstack((out,a[-1:]))
print(out)

输出:

[   2.   15.   40.   65.   66.   91.  116.  141.  166.  180.]

此解决方案假设 a 是平面数组。

编辑:之后，我注意到我的解决方案生成 dtype float64 的 np.array，即不同的输入 - 根据您的使用情况，这可能会或不会成为问题。对于出现问题的情况，我将代码修改为以下形式:

import numpy as np
a = np.array([2,15,66,180])
print(a.dtype) #int64
out = np.hstack([np.arange(a[i],a[i+1],25) for i in range(len(a)-1)]+[a[-1:]])
print(out) #[  2  15  40  65  66  91 116 141 166 180]
print(out.dtype) #int64

代码也更短，但我保留了原始代码，因为它可能更容易理解。