如何在python中用二进制值替换列值？

Question

如何在python中将列值替换为二进制值，例如我有这个表：

age | height
------------
20  | 160
30  | 173
21  | 174
32  | 165
13  | 140
26  | 160
40  | 164
10  | 131
11  | 140

我想将每列的值更改为二进制值，例如对于年龄列，我想将值更改为“<30”和“>=30”，并将高度列更改为“<160”和“>= 160"

我已经尝试使用：

mask = df[0] < 30
df.loc[mask, 0] = "<30"

mask2 = df[0] >= 30
df.loc[mask2, 0] = ">=30"

但我希望它更简单

Answer 1

您可以尝试以下操作吗：

import numpy as np
import pandas as pd

df = pd.read_csv(<yourcsvfile>)
df['age'] = np.where(df['age'] >= 30, ">=30", "<=30")
df['height'] = np.where(df['height'] >= 160, ">=160", "<=160")