         a   b
0        1   4
1        4  16
2        9  25

[3 rows x 2 columns]