python - 以相等的增量向零扩展数组

Question

我有各种数据框，每个数据框都有不同的深度范围。

对于更复杂的计算（这是此处发布的问题的一个片段：Curvefitting for each column in Pandas + extrapolate values），我需要编写一个函数，因此它将扩展深度列/数组以获得相等的增量dz（在此情况 0.5) 朝向零（表面）。这里缺少的配额是 0.15、0.65 和 1.15

import numpy as np

depth = np.array([1.65, 2.15, 2.65, 3.15, 3.65, 4.15, 4.65, 5.15, 5.65, 6.15, 6.65, 7.15, 7.65, 8.15, 8.65])

任何想法如何编写一个函数，以便每次都针对不同的深度范围（即取决于不同的最小值）执行它？

标签： pythonpandasnumpy

Answer 1

我做的一个非常简单的解决方案是：

depth_min = np.min(depth)
step = 0.5
missing_vals = np.arange(depth_min - step, 0, -step)[::-1]
depth_tot = np.concatenate((missing_vals, depth), axis=0)

我确信存在更好的方法