Назначьте int строковому значению после открытия CSV-файла в pandas

Question

Я новичок в python и только что научился пользоваться пандами.У меня есть CSV-файл с именем Data.csv, который имеет значение A или B в последнем столбце

import pandas as pd
data = pd.read_csv('filelocation/Data.csv', header = None)

last_col = data.iloc[0:208, 60]
print(last_col)

Последний столбец выглядит так:

A
B
B
A
A

Моя проблема сейчасчто я хочу назначить целое число для строк (то есть A = 0 и B = 1), чтобы при повторной загрузке последнего столбца результаты были:

0
1
1
0
0

Спасибо за вашу помощь!

Answer 1

Определите свое отображение в словаре и передайте .replace(), например, так:

In[2]: import pandas as pd
  ...: 
  ...: df = pd.DataFrame(['A', 'B'])
  ...: 
  ...: # inspect before
  ...: df
  ...: 
Out[2]: 
   0
0  A
1  B
In[3]: # define mapping for the translation
  ...: mapping = {
  ...:     'A': 0,
  ...:     'B': 1,
  ...: }
  ...: 
  ...: # apply the mapping
  ...: df[0] = df[0].replace(mapping)
  ...: 
  ...: # inspect after
  ...: df
  ...: 
Out[3]: 
   0
0  0
1  1