            a    b
0         4.0  1.0
1         9.0  4.0
2        16.0  9.0

[3 rows x 2 columns]