          a  b
0         4  1
1         9  4
2        16  9

[3 rows x 2 columns]