使用np.where从DataFrame赋值是一种条件赋值的方法。np.where函数可以根据条件选择性地对DataFrame进行赋值操作。
具体步骤如下:
import numpy as np
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3, 4, 5],
'B': [6, 7, 8, 9, 10]})
df['C'] = np.where(df['A'] > 3, 'High', 'Low')
上述代码中,np.where函数的第一个参数是条件,第二个参数是满足条件时的赋值,第三个参数是不满足条件时的赋值。在这个例子中,如果'A'列的值大于3,则'C'列的值为'High',否则为'Low'。
print(df)
输出结果如下:
A B C
0 1 6 Low
1 2 7 Low
2 3 8 Low
3 4 9 High
4 5 10 High
使用np.where从DataFrame赋值的优势是可以根据条件快速进行批量赋值操作,提高了代码的简洁性和效率。
应用场景:
推荐的腾讯云相关产品和产品介绍链接地址:
以上是关于如何使用np.where从DataFrame赋值的完善且全面的答案。
领取专属 10元无门槛券
手把手带您无忧上云