操纵/重新编码熊猫数据框

Question

在Pandas中进行一些简单的数据操作，不确定如何按照以下定义的逻辑进行操作。我正在尝试根据此示例思想中的列A中的值更改列C中的值。有建议吗？

df = pd.DataFrame({'A': [0, 1, 2, 3, 4],
                   'B': [5, 6, 7, 8, 9],
                   'C': ['a', 'b', 'c', 'd', 'e']})
if df['A'] < 2:
    df['C'] = "Small"
else:
    df['C'] = "Big"

Answer 1

大熊猫的工作条件不同。你可以尝试

import numpy as np
df['C'] = np.where(df.A < 2, 'small','big')
df

输出

   A  B      C
0  0  5  small
1  1  6  small
2  2  7    big
3  3  8    big
4  4  9    big