python - 使用平均方法下采样

Question

我有一个 2 维数据框，我想按行进行下采样。因此，如果我有一个 100x2000 数据帧，我希望它是 100x500 。我尝试过滚动平均值，但输出的尺寸是基于列重新采样的，并且仍然是原始形状。尝试寻找其他方法我也发现了 signal.resample 但我相信它不适合我的场景。如果我有 [2,4,6,8,10,12,14,16,18] 我想将其作为输出 [4,10,16] 这意味着 3 个数字的滑动窗口，同时计算平均值

有什么想法吗？

谢谢

Answer 1

使用numpy.convolve:

import numpy as np

arr = np.array([2,4,6,8,10,12,14,16,18])
n = 3
window = (1.0 / n) * np.ones(n,)
res = np.convolve(arr, window, mode='valid')[::n]

对于 2 x N 数组:

from scipy import signal

arr = np.array([[2,4,6,8,10,12,14,16,18], [12,14,16,18,110,112,114,116,118]])
window = (1.0 / n) * np.ones((1, n))
#res = np.convolve(arr, window, mode='valid')[::n]
res = signal.convolve2d(arr, window, mode='valid')[:, ::n]