如何将熊猫数据框中的字符串条目转换为整数？

Question

我正在使用的pandas数据框有一个包含字符串条目的列，我希望将其转换为整数。

该列称为diagnosis，每行的值分别为M或B。我想使用以下代码将所有M转换为1并将所有B转换为0

for row in data['diagnosis']:
  if row == 'M':
    row = 1
  else:
    row = 0

但是它不会改变任何东西吗？

Answer 1

您可以使用map来完成

data['diagnosis'] = data['diagnosis'].map({'M':1,'B':0})