Pandas sort staircase date-CodePudding

I have a set of data, where I predict the amount of fuel I need around 10 weeks ahead. I have it all set up in a single dataframe presented as staircase date. This means, the closer I come to the last entry for a week the more accurate the values get. I want to cut all missing values and ignore the exact date so I can just look at my predictions in relation to the distance of the predicted week.

Input dataframe:

Index 2020-01   2020-02  2020-03  2020-04  2020-05  2020-06
1.         10        10        5        0        0        0
2.          0         5        5       10        0        0
3.          0         0       10        4        3        0
4.          0         0        0        1        7        6

Outcome should be:

Index      W1        W2      W3
1.         10        10       5     
2.          5        5       10   
3.         10        4        3        
4.          1        7        6

Many Thanks in advance

CodePudding user response：

You could replace the zeros with NaNs and reset the Series per row:

df2 = (
 df.replace(0,float('nan'))
   .apply(lambda s: s.dropna().reset_index(drop=True), axis=1)
   .astype(int) 
)

df2.columns = df2.columns.map(lambda x: f'W{x 1}')

output:

     W1  W2  W3
1.0  10  10   5
2.0   5   5  10
3.0  10   4   3
4.0   1   7   6

CodePudding user response：

Use justify function for remove shift non 0 values, last remove columns filled only 0 values:

c = [f'W{x   1}' for x, _ in enumerate(df.columns)]
df = pd.DataFrame(justify(df.to_numpy()), index=df.index, columns=c)
df = df.loc[:, df.ne(0).any()]
print (df)
       W1  W2  W3
Index            
1.0    10  10   5
2.0     5   5  10
3.0    10   4   3
4.0     1   7   6

##https://stackoverflow.com/a/44559180/2901002
def justify(a, invalid_val=0, axis=1, side='left'):    
    """
    Justifies a 2D array

    Parameters
    ----------
    A : ndarray
        Input array to be justified
    axis : int
        Axis along which justification is to be made
    side : str
        Direction of justification. It could be 'left', 'right', 'up', 'down'
        It should be 'left' or 'right' for axis=1 and 'up' or 'down' for axis=0.

    """

    if invalid_val is np.nan:
        mask = ~np.isnan(a)
    else:
        mask = a!=invalid_val
    justified_mask = np.sort(mask,axis=axis)
    if (side=='up') | (side=='left'):
        justified_mask = np.flip(justified_mask,axis=axis)
    out = np.full(a.shape, invalid_val) 
    if axis==1:
        out[justified_mask] = a[mask]
    else:
        out.T[justified_mask.T] = a.T[mask.T]
    return out

CodePudding user response：

Using a custom function and apply would be the most straightforward and easily understood way:

def merge_row(row):
    vals = [v for v in row.values if v != 0]
    return pd.Series({f'W{i}': v for i, v in enumerate(vals)})

df.apply(merge_row, axis=1)

Result:

     W0  W1  W2
Index           
1.0  10  10   5
2.0   5   5  10
3.0  10   4   3
4.0   1   7   6

CodePudding user response：

You can use numpy to sort by the 0/non-0 state and rebuild a DataFrame:

a = df.to_numpy()
b = a==0
idx = np.argsort(b, axis=1)
n_cols = b.sum(1).max()
pd.DataFrame(np.take_along_axis(a, idx, axis=1)[:, :n_cols],
             columns=[f'W{i}' for i in np.arange(n_cols) 1],
             index=df.index
            )

output:

     W1  W2  W3
1.0  10  10   5
2.0   5   5  10
3.0  10   4   3
4.0   1   7   6