python - 在 Pandas 中的另一列的基础上创建新列

Question

我在 Dataframe 中有一个名为 STATUS 的列，其中包含值“AMEND ORDER”或者为空。我需要在该列的基础上创建一个 FLAG，如果它的值包含 AMEND 单词，则将 FLAG 设置为 1 或 0。我尝试使用下面的代码，但它出错了。

t[t['STATUS'].str.contains('AMEND')] 

Answer 1

如果为空则缺少值或 None 通过 | 链接另一个掩码进行按位 OR 并转换为整数进行 True -> 1, False -> 0 映射:

t['FLAG'] = (t['STATUS'].str.contains('AMEND') | t['STATUS'].isna()).astype(int)

如果为空，则为空字符串，将值与 Series.eq 进行比较:

t['FLAG'] = (t['STATUS'].str.contains('AMEND') | t['STATUS'].eq('')).astype(int)