Square of each element of a column in pandas

Nothing wrong with the accepted answer, there is also:

df = pd.DataFrame({'a': range(0,100)})
np.square(df)
np.power(df, 2)

Which is ever so slightly faster:

In [11]: %timeit df ** 2
10000 loops, best of 3: 95.9 µs per loop

In [13]: %timeit np.square(df)
10000 loops, best of 3: 85 µs per loop

In [15]: %timeit np.power(df, 2)
10000 loops, best of 3: 85.6 µs per loop

>>> import pandas as pd
>>> df = pd.DataFrame([[1,2],[3,4]], columns=list('ab'))
>>> df
   a  b
0  1  2
1  3  4
>>> df['c'] = df['b']**2
>>> df
   a  b   c
0  1  2   4
1  3  4  16

Tags:

Python

Pandas