如果該元素爲0，則在該列中查找上方和下方元素的平均值 - Pandas DataFrame

Question

我想使用來自另一個數據幀的相同值創建新數據幀，除非存在0值。如果有一個0值，我想在前後找到入口的平均值。

例如：

df = A B C 
     5 2 1 
     3 4 5 
     2 1 0 
     6 8 7 

我想結果看起來像下面的DF：

df_new = A B C 
     5 2 1 
     3 4 5 
     2 1 6 
     6 8 7 

Answer 1

import pandas as pd 
import numpy as np 

df = pd.DataFrame({'A':[5, 3, 2, 6], 'B':[2, 4, 1, 8], 'C':[1, 5, 0, 7]}) 

Nrows = len(df) 

def run(col): 
     originalValues = list(df[col]) 
     values = list(np.where(np.array(list(df[col])) == 0)[0]) 
     indices2replace = filter(lambda x: x > 0 and x < Nrows, values) 
     for index in indices2replace: 
       originalValues[index] = 0.5 * (originalValues[index+1] + originalValues[index-1]) 
     return originalValues 


newDF = pd.DataFrame(map(lambda x: run(x) , df.columns)).transpose()