преобразование ул в int

Question

У меня есть текстовый файл, и я пытаюсь его обработать. в конце я должен преобразовать одну из переменных из str в int. если я выведу переменную с помощью этой команды:

print(y_test.iloc[:,1])

, я получу следующие результаты:

0     B
1     B
2     A
3     A
4     A
5     A
6     A
7     A
8     B
9     B
10    B
11    B
12    B
13    B
Name: group, dtype: object

и получу тип этой переменной:

print(type(y_test.iloc[:,1]))

Я бы получил этот вывод:

<class 'pandas.core.series.Series'>

, но мой ожидаемый вывод такой (от A до 0 и от B до 1):

0     0
1     0
2     1
3     1
4     1
5     1
6     1
7     1
8     0
9     0
10    0
11    0
12    0
13    0

, чтобы сделать это, я попыталсяследующий код:

y2 = int(y_test.iloc[:,1])

но не вернул, что я хочу. ты знаешь как это сделать?

Answer 1

Чтобы получить A как 1 и B (или что-либо еще) как 0, вы можете использовать ниже. Он преобразует все A в True, которые затем преобразуются в целочисленное значение True, которое равно 1.

(y_test.iloc[:,1] == 'A').astype(int)

Answer 2

Вы можете использовать метод replace():

y_test.iloc[:, 1].replace({'B': 0, 'A': 1})