pandas习题 068：按条件计算增加两个新列
看过来
《pandas 教程》 持续更新中，提供建议、纠错、催更等加作者微信: gairuo123（备注：pandas教程）和关注公众号「盖若」ID: gairuo。跟作者学习，请进入 Python学习课程。欢迎关注作者出版的书籍：《深入浅出Pandas》 和 《Python之光》。
（编码题）以下有一个名为 df 的列，需求为增加两个新列，新列的取值逻辑为：
- 新增 B2 列，如果 B 列对应值为缺失值，值为 1，否则为 0；
- 新增 C2 列，如果 C 列对应值大于 10，值为 1，否则为 0。
import pandas as pd
import numpy as np
data = {
'A': [1, 2, np.nan, 4],
'B': [5, np.nan, 7, 8],
'C': [9, 10, 11, np.nan]
}
df = pd.DataFrame(data)
df
'''
A B C
0 1.0 5.0 9.0
1 2.0 NaN 10.0
2 NaN 7.0 11.0
3 4.0 8.0 NaN
'''
参考答案
以下为 Python 代码：
(
df.assign(B2=lambda d: np.where(d.B.isna(), 1, 0))
.assign(C2=lambda d: np.where(d.C > 10, 1, 0))
)
'''
A B C B2 C2
0 1.0 5.0 9.0 0 0
1 2.0 NaN 10.0 1 0
2 NaN 7.0 11.0 0 1
3 4.0 8.0 NaN 0 0
'''
解析
查看相关链接中的知识。
（完）
相关内容
- pandas 练习题 03 2024-10-22 20:18:16
- NumPy np.where() 按条件替换值 2024-12-01 17:27:21
更新时间：2024-12-01 20:07:53 标签：pandas python 赋值 增加
