用数字替换2d数组中的字符串值

Question

Imported Data

您好，

我在python中使用read_csv函数从url导入数据。我导入的数据包含一列字符/字符串，其他列由数字组成。有没有办法用数字替换第一列？例如，将M替换为1，将F替换为2，将I替换为3？我正在考虑使用条件语句进行某种循环，但我并不真正了解python的正确语法。

    While i less than number of rows
       If X[i][0] == 'M' 
       X[i][0] = 1

       Else X[i][0] =='F'
       X[i][0] == 2
      i=i+1

Answer 1

从pandas函数名称和图像中数据的格式设置猜测read_csv()为库。

使用pandas，您不希望使用用Python代码编写的显式循环，而是使用pandas提供的工具。在这种情况下，数据框上有map()方法。

import pandas
df = pandas.read_csv('test.csv')
df.sex = df.sex.map({'M': 1, 'F': 2, 'I': 3})
print(df)
print(df.dtypes)