如何基于“描述列”创建“类别”列

时间:2019-01-27 21:10:47

标签: python sql dataframe

我正在使用“描述”列处理银行交易数据集,该列包含描述交易的单个短语。我想将数据集中的类似交易归类。

例如,如果事务1的描述为“ Roger-B111”,事务2的描述为“ Internet bill 293290”,那么我想将它们的描述都更改为“电信”。

我要寻找的最终结果是一个名为“交易类别”的新列,该列将具有不同的类别级别(食品,电信,账单)。

有什么最好的方法吗?

1 个答案:

答案 0 :(得分:0)

您可以使用case,但必须弄清楚逻辑:

select t.*,
       (case when description in ('Roger - B111', 'Internet bill 293290')
             then 'Telecommunication'
             . . .
        end) as category
from t;