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

时间:2018-09-24 11:46:51

标签: python pandas numpy

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

对于更复杂的计算(这是一个问题的片段,在此处发布:Curve fitting 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])

有什么想法如何编写函数,以便每次在不同的深度范围内(即取决于变化的最小值)执行该函数吗?

1 个答案:

答案 0 :(得分:0)

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

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)

我敢肯定还有更好的方法