Pandas groupby average after slicing over non-zero values across remaining groups

Question

Hi Consider the following dataframe

import pandas as pd
import numpy as np
a = pd.DataFrame(np.array([[1,1,1,1], [1,1,2,0], [1,2,1,1], [1,2,2,0], [2,1,1,0], [