我在徘徊如何添加一个新列,其中包含从其他列计算的值。
让我们说“小”'是一个值小于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()的内容,但我仍然没有得到它。
非常感谢帮助谢谢!
答案 0 :(得分:2)
您可以将布尔值掩码转换为integer
,然后转换为string
:
rawData['small'] = (rawData['value'] < 5).astype(int).astype(str)