创建一个具有不同值的列，具体取决于熊猫另一列中的值

Question

考虑以下（虚拟）熊猫数据集：

如何用Python构造“类型”列？

为澄清起见，必须以以下方式构造该值：六个最高月份值的Type为'F'，其余行的Type为'A'。这是因为“月份”列可能并不总是介于0到10之间，例如如果Month介于0-15之间，则10-15个月的类型为'F'。

Answer 1

使用numpy更好：

month_limit = df['Month'].max()-5

df['type'] = np.where(df['Month'] >= month_limit, 'F', 'A')

Answer 2

希望这会起作用：

month_limit = df['Month'].max()-5
df['Type'] = ['F' if x >= month_limit else 'A' for x in df['Month']]