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

Question

这个问题在这里已经有了答案:





Pandas conditional creation of a series/dataframe column

(10 个回答)


12 个月前关闭。




在 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

Pandas 的条件不同。你可以试试

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