如何將 pandas dataframe 中的字符串條目轉換為整數？

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})