Передискретизация данных в целевой сетке
Есть ли эффективный способ решить следующую проблему?
Представьте, что у нас есть эти 2 списка. Они представляют неравномерно выбранные данные:
MeasTimes = [0, 2, 4, 6, 8, 11, 12, 14, 18, 20, 22]
MeasValues = [-1, 0, 1, 0, -1, 0.5, 1, 0, 0, 1, 0]
Как можно интерполировать данные этого измерения в целевую сетку? Пример:
TargetTimes = [0, 5, 10, 15, 18, 20] # given
2 ответа
Решение
meas_times = [0, 2, 4, 6, 8, 11, 12, 14, 18, 20, 22]
meas_values = [-1, 0, 1, 0, -1, 0.5, 1, 0, 0, 1, 0]
def get_value(target_value):
if target_value in meas_times:
return meas_values[meas_times.index(target_value)]
else:
upper_index, upper_time = next((i, e) for i, e in enumerate(meas_times) if e > target_value)
lower_value, upper_value = meas_values[upper_index - 1], meas_values[upper_index]
abs_dt = float(abs(lower_value) + abs(upper_value)) / (upper_time - meas_times[upper_index - 1])
if upper_value > lower_value:
return lower_value + abs_dt * (target_value - meas_times[upper_index - 1])
else:
return lower_value - abs_dt * (target_value - meas_times[upper_index - 1])
target_times = [0, 5, 10, 15, 18, 20]
print map(get_value, target_times)
# [-1, 0.5, 0.0, 0.0, 0, 1]
Я думаю, что в разы вы дали неправильные данные ( 5 и 10)
Кодекс
MeasValuesAtTargetTimes=[MeasValues[MeasTimes.index(i)] for i in TargetTimes]