如何使用Pandas(相当于SQL中的“Like”运算符)获取列的值

2024-03-29 13:09:34 发布

您现在位置:Python中文网/ 问答频道 /正文

我可以在dataframe中获得特定的列值,比如类似SQL的运算符,它可以找到任何值,然后计算值以将其存储在新列中。这是我的数据帧的代码

    import pandas as pd
    
    dataku = pd.DataFrame()
    
    dataku['CIF'] = ['789', '290', '789', '789','290']
    dataku['NAMA'] = ['de','ra','de','de','ra']
    dataku['SALDO'] = [100,500,800,200,500]
    dataku ['PRODUK']=['tabungan','deposito','deposito','tabungan','deposito usd']

dataku.groupby(['CIF','NAMA','PRODUK']).agg({'SALDO':'sum', 'PRODUK':'count'}).rename(columns={'SALDO':'TOTAL SALDO','PRODUK':'TOTAL PRODUK'})

我希望新数据帧的结果如下

CIF NAMA PRODUK   TOTAL_SALDO  TOTAL_PRODUK  GT_SALDO  GT_PRODUk
290  ra  deposito         500             1      1000         2
         deposito usd     500             1                      
789  de  tabungan         300             2       300         2
         deposito         800             1       800         1

我如何获得GT_SALDO列和GT_PRODUK的值(如上表所示)作为最终结果