python - 如何使一列中的值依赖于另一列？

Question

我有一列名为“A”，只有值 0 或 1。我还有另一列名为“B”。如果 A 列值 = 0，我希望 B 列值等于“狗”。如果 A 列值 = 1，我希望 B 列值等于“cat”。

示例 DataFrame 列：

print(df)
   A
0  0
1  1

无论如何都可以在没有 for 循环的情况下填充 B 列吗？

期望：

print(df)
   A  B
0  0  Cat
1  1  Dog

谢谢

标签： pythondataframe

Answer 1

可以简单地尝试下面使用map...

样本数据

print(df)
   A
0  0
1  1
2  0
3  1
4  1
5  1
6  0
7  0
8  1

结果：

df['B'] = df['A'].map({0:'Cat', 1:'Dog'})
print(df)
   A    B
0  0  Cat
1  1  Dog
2  0  Cat
3  1  Dog
4  1  Dog
5  1  Dog
6  0  Cat
7  0  Cat
8  1  Dog