如何将 color 列添加到以下数据帧，以便 color='green' 如果 set == 'z'，否则 color='red' ？
Type Set 1 A Z 2 B Z 3 B X 4 C Y
如果您只有两个选择，请使用 np.where：
df['color'] = np.where(df['set']=='z', 'green', 'red')
例如，
import pandas as pd
import numpy as np
df = pd.dataframe({'type':list('abbc'), 'set':list('zzxy')})
df['color'] = np.where(df['set']=='z', 'green', 'red')
print(df)产量
set type color 0 z a green 1 z b green 2 x b red 3 y c red
如果您有两个以上的条件，请使用 np.select。例如，如果您希望 color 为
然后使用
df = pd.dataframe({'type':list('abbc'), 'set':list('zzxy')})
conditions = [
(df['set'] == 'z') & (df['type'] == 'a'),
(df['set'] == 'z') & (df['type'] == 'b'),
(df['type'] == 'b')]
choices = ['yellow', 'blue', 'purple']
df['color'] = np.select(conditions, choices, default='black')
print(df)产生
Set Type color 0 Z A yellow 1 Z B blue 2 X B purple 3 Y C black
以上就是如何创建一个新列，其中的值是根据现有列选择的？的详细内容，更多请关注php中文网其它相关文章！
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移，垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是，许多工具可以让 Windows 保持平稳运行。
广告![ThinkPHP5快速开发企业站点[全程实录]](https://img.php.cn/upload/course/000/000/068/6253d918a3ce7278.png)
收藏
收藏
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号
482
