首先新建一个Dataframe

import pandas as pd

df = pd.DataFrame({'a':[1,2,3],'b':[1,2,3]})

如下：
   a  b
0  1  1
1  2  2
2  3  3

1、修改列名a，b为A、B。

df.columns = ['A','B']

2、只修改列名a为A

df.rename(columns={'a':'A'})

如你想了解更多关于python和spark以及机器学习的知识，可以关注下面的公众号我们一起沟通交流。