python - 如何检查系列的符号是否符合给定的符号字符串？

Question

例如我有一个系列如下，

ts = pd.Series([-1,-2.4,5,6,7, -4, -8])

我想知道是否有Pythonic方法来根据符号列表检查ts的符号，例如，

sign = '++++---' # returns False

同时

sign = '--+++--' # returns True

Answer 1

此解决方案需要一些 numpy 函数，但由于您使用 pandas 来处理数据系列，因此这对您来说可能不是问题。

import pandas as pd
import numpy as np

values = pd.Series([-1, -2.4, 5, 6, 7, -4, -8, 0])
sign_str = "--+++--0"

sign_map = {
    "+" : 1,
    "0" : 0,
    "-" : -1
}

expected_signs = list(map(sign_map.get, sign_str))
observed_signs = np.sign(values)
np.all(expected_signs == observed_signs)