根據其中的列名和值修改 Dataframe

Question

我有關注 dataframe

一個	乙	C	D	乙
-0.1	0	0.2	0	4
0	0	-1	-2	5

我想要一個 output 如下基於 A,B 屬於 X 類和 C,D,E 屬於 Y 類 -

一個	乙	C	D	乙
-貓X	0	+貓Y	0	+貓Y
0	0	-貓Y	-貓Y	+貓Y

它基本上檢查列名並分配類別並檢查值以分配符號。 在 Python 中是否有任何簡單的方法可以做到這一點？ 我正在使用 COLAB，所以可能有最新版本。

Answer 1

一種方法可能是只選擇您想要的任何列，然后執行如下檢查：

import numpy as np

for col in ['A','B']:
    df[col] = [x if x == 0 else '-Cat X' if x < 0 else '+ Cat X' for x in df[col]]

for col in ['C','D','E']:
    df[col] = [x if x == 0 else '-Cat Y' if x < 0 else '+ Cat Y' for x in df[col]]

這應該會產生所需的結果。