使用从其他列

时间:2018-03-24 09:17:13

标签: python pandas numpy

我在徘徊如何添加一个新列,其中包含从其他列计算的值。

让我们说“小”'是一个值小于5的那个。我想在一个名为' small'的新列中记录这些信息。如果该列为短,则此列的值为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()的内容,但我仍然没有得到它。

非常感谢帮助谢谢!

1 个答案:

答案 0 :(得分:2)

您可以将布尔值掩码转换为integer,然后转换为string

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