python - python 中的一个命令行可以自动化代码

Question

我是Python新手。我用它来进行脑电图机器学习。这是我提取“特征”的函数。有办法改善吗？我的意思是，我不想每次需要其他范围时都改变频率。所以，在 B 部分你会看到我的尝试。

A 部分

def computePowerBands(f, amp):
    return (np.mean(amp[(f >= 0.5)*(f <= 4.5)]),
            np.mean(amp[(f >= 4.5)*(f <= 8.5)]),
            np.mean(amp[(f >= 8.5)*(f <= 11.5)]),
            np.mean(amp[(f >= 11.5)*(f <= 15.5)]),
            np.mean(amp[(f >= 15.5)*(f <= 32.5)]) )

B 部分

def computePowerBands(f, amp, fce):
    return (np.mean(amp[k * k for k in fce]))

有什么办法可以做到这一点吗？

Answer 1

我不完全确定您想要实现什么，因为它确实是令人困惑的代码块，但通过简单的代码重构，您可以通过对列表对其进行参数化:

def computePowerBands(f, amp, fce):
    return [np.mean(amp[(f >= low)*(f <= up)]) for low, up in fce]

并调用它

computePowerBands(f, amp, [(0.5, 4.5), (4.5, 8.5)])

等等。

或者如果以下乐队总是共享末端:

def computePowerBands(f, amp, fce):
    return [np.mean(amp[(f >= fce[fid])*(f <= fce[fid+1])])
            for fid in range(len(fce)-1)]

并调用它

computePowerBands(f, amp, [0.5, 4.5, 8.5])