           a     b
0        1.0   4.0
1        4.0  16.0
2        9.0  25.0

[3 rows x 2 columns]