python - 操作/重新编码 Pandas 数据框

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"

标签： pythonpandas

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