python - 如何更新具有多个条件的列，从另一列 Python Pandas 中获取相应的值

Question

输出更像是下面的 SQL 语句。

   UPDATE table_A SET final=(cs+fhfa+sz)/3 WHERE cs IS NOT NULL AND fhfa IS NOT NULL AND sz IS NOT NULL;

这里cs+fhfa+sz都是sql表(和dataframe)中的各个列

如果我想把这个 SQL 语句转换为 python 中的 pandas 操作，这将更像是:

   df['div_3'] = (df.cs+df.fhfa+df.sz) /3
   df['final'] = df.loc[(df['cs'] != None) & (df['fhfa'] != None) & (df['sz'] != None) ] = df['div_3']

但这并不能保证“相应的值”最终被放入。如何实现？？

我真的需要用 3 列的总和创建另一列 div_3 吗？这可以在不创建另一个列的情况下完成吗？

Answer 1

过滤 pd.Series.notnull 并调用 mean。

c = ['cs', 'fhfa', 'sz']
df['final'] = df[df[c].notnull().all(1)][c].mean(1)