read some column in csv data using pandas

Question

i have data csv

word,centroid
she,1
great,0
good,3
mother,2
father,2
After,4
before,4
.....

I just want to see the centroid

my code is:

df = pd.read_csv('data/hasil_cluster.csv',encoding = "ISO-8859-1") 
print(df['centroid'])

and error:

KeyError: 'centroid'

Answer 1

There is ; after centroid :

print (df.columns.tolist())
['word', 'centroid;'] 

You can remove all ; before and after column names:

df.columns = df.columns.str.strip(';')

Or use rename :

df = df.rename(columns={'centroid;':'centroid'})

Or reassign columns names:

df.columns = ['word', 'centroid']