how to clip pandas dataframe column-wise?

Question

I have

In [67]: a
Out[67]:

   0  1  2
0  1  2  3
1  4  5  6

when I run

In [69]: a.clip(lower=[1.5,2.5,3.5],axis=1)

I got

ValueError: other must be the same shape as self when an ndarray

Is that expected? I was expecting to get something like:

Out[72]:

     0    1    2
0  1.5  2.5  3.5
1  4.0  5.0  6.0

Answer 1

Instead of a numpy array, you can use a Series so the labels are aligned:

df
Out: 
   A  B
0  1  4
1  2  5
2  3  6

df.clip(lower=pd.Series({'A': 2.5, 'B': 4.5}), axis=1)
Out: 
     A    B
0  2.5  4.5
1  2.5  5.0
2  3.0  6.0