Pandas：如何过滤一列的行以获得每两个连续单元格之间的交替符号？

Question

我有一个看起来像这样的数据框

df = pd.DataFrame({"a":[-0.1, -0.2, 0.2, -0.1, 0.1, 0.5, 0.6]})
df
    a
0   -0.1
1   -0.2
2   0.2
3   -0.1
4   0.1
5   0.5
6   0.6

我想过滤 a 列a行，使两个连续单元格的符号交替出现。

例如，如果我想在第一个单元格中以正数开头，那么过滤后的 dataframe 将如下所示

    a
2   0.2
3   -0.1
4   0.1

或者，如果我想以负数开头，则 dataframe 将如下所示

    a
0   -0.1
2   0.2
3   -0.1
4   0.1

我怎样才能做到这一点？

谢谢！

更新：

到目前为止，我对以正数开头的 dataframe 的解决方案是

df_long_first = pd.DataFrame(columns = ['a'])
for i in range(len(df)):
    cur_val = df['a'].iloc[i]
    cur_sign = np.where(cur_val > 0, "p", "n")
    if i == 0:
        df_long_first = df_long_first.append({"a" : cur_val}, ignore_index = True)
    else:
        last_sign = np.where(df_long_first['a'].iloc[-1] > 0, "p", "n")
        if last_sign == cur_sign:
            continue
        else:
            df_long_first = df_long_first.append({"a" : cur_val}, ignore_index = True)
            
        
df_long_first

我想知道是否有更优雅的方法来做到这一点？

Answer 1

这是我目前能想到的最好的。 function 采用一维数组并迭代其索引和值。 它会将给定符号删除/添加到具有偶数/奇数索引的值。 默认情况下，如果值具有符号和偶数索引，它会删除符号。 start_no_sym 默认参数可用于更改模式。 它返回一个一维数组。

def alternate(col, symbol, start_no_sym=True):
    alternated = []

    # no symbol, symbol pattern  
    if start_no_sym:
        for idx, val in enumerate(col):
            # remove symbol from even index vals
            if idx % 2 == 0 and symbol in val: alternated.append(val[1:])
            # if odd idx val has no symbol, add it
            elif idx % 2 != 0 and symbol not in val: alternated.append(f'{symbol}{val}')
            else: alternated.append(val)
    
    # symbol, no symbol pattern
    else:
        for idx, val in enumerate(col):
            # add symbol to even index vals with no symbol
            if idx % 2 == 0 and symbol not in val: alternated.append(f'{symbol}{val}')
            # if odd index val has a symbol, remove it
            elif idx % 2 != 0 and symbol in val: alternated.append(val[1:])
            else: alternated.append(val)

    return alternated

# test
print(alternate(['<3', "9", '<0', '<6', '<2', '8', '1', '6'], '<'))