python - 用从其他列计算出的值添加新列

我在徘徊如何添加具有从其他列计算出的值的新列。

假设“ small”是一个值小于5的值。我想将此信息记录在一个名为“ small”的新列中，该列的值如果为short则为1，否则为0。

我知道你可以像这样用numpy做到这一点：

import pandas as pd
import numpy as np

rawData = pd.read_csv('data.csv', encoding = 'ISO-8859-1')
rawData['small'] = np.where(rawData['value'] < 5, '1', '0')

不使用numpy怎么做相同的操作？
我已经读过有关numpy.where（）的信息，但我仍然不明白。

非常感谢您的帮助！

最佳答案

您可以将布尔掩码转换为integer，然后转换为string：

rawData['small'] = (rawData['value'] < 5).astype(int).astype(str)