如何创建一个新列，其中的值是根据现有列选择的？

百变鹏仔 1天前 #Python

文章标签创建一个

问题内容

如何将 color 列添加到以下数据帧，以便 color='green' 如果 set == 'z'，否则 color='red' ？

Type  Set1     A    Z2     B    Z           3     B    X4     C    Y

正确答案

如果您只有两个选择，请使用 np.where：

df['color'] = np.where(df['set']=='z', 'green', 'red')

例如，

import pandas as pdimport numpy as npdf = pd.dataframe({'type':list('abbc'), 'set':list('zzxy')})df['color'] = np.where(df['set']=='z', 'green', 'red')print(df)

产量

set type  color0   z    a  green1   z    b  green2   x    b    red3   y    c    red

如果您有两个以上的条件，请使用 np.select。例如，如果您希望 color 为

然后使用

df = pd.dataframe({'type':list('abbc'), 'set':list('zzxy')})conditions = [    (df['set'] == 'z') &amp; (df['type'] == 'a'),    (df['set'] == 'z') &amp; (df['type'] == 'b'),    (df['type'] == 'b')]choices = ['yellow', 'blue', 'purple']df['color'] = np.select(conditions, choices, default='black')print(df)

产生

Set Type   color0   Z    A  yellow1   Z    B    blue2   X    B  purple3   Y    C   black

文章推荐

如何创建一个新列，其中的值是根据现有列选择的？

正确答案

如何用Python编写桶排序算法？

python语言%表示什么意思

python如何随机生成100内的10个整数

如何使用Python实现DBSCAN聚类算法？

如何使用Python实现梯度下降算法？