Question

我有一个带有分类变量的数据框。我想使用以下逻辑将它们转换为数值：

我有2个列表，一个包含列中不同的类别值，第二个列表包含每个类别的值。现在，我需要映射这些值来代替那些分类值。

例如：

List_A = ['A'，'B'，'C'，'D'，'E']

List_B = [3,2,1,1,2]

我需要将A替换为3，将B替换为2，将C和D替换为1，将E替换为2。

有什么办法可以在Python中做到这一点。

我可以通过应用多个for循环来做到这一点，但是我正在寻找一些更简单的方法或直接函数（如果有）。

非常感谢您的帮助。

Answer 1

创建映射字典

List_A = ['A','B','C','D','E',]

List_B = [3,2,1,1,2]
d=dict(zip(List_A, List_B))

new_list=['A','B','C','D','E','A','B']
new_mapped_list=[d[v] for v in new_list if v in d]
new_mapped_list

或定义一个函数并使用map

List_A = ['A','B','C','D','E',]

List_B = [3,2,1,1,2]

d=dict(zip(List_A, List_B))

def mapper(value):
    if value in d:
        return d[value]
    return None

new_list=['A','B','C','D','E','A','B']
map(mapper,new_list)

Answer 2

假设df是您的数据框，而“ Category”是保存类别的列的名称：

df[df.Category == "A"] = 3,2, 1, 1, 2
df[(df.Category == "B") | (df.Category == "E") ] = 2
df[(df.Category == "C") | (df.Category == "D") ] = 1

Answer 3

如果只需要将一个列表中的值替换为另一个列表中的值，并且结构类似于您所说的结构。两个列表，相同的长度和相同的位置，那么您只需要这样：

list_a = []
list_a = list_b

更复杂的解决方案是这样的，它具有一个可以创建可在其他列表上使用的字典的函数：

# we make a function
def convert_list(ls_a,ls_b):
    dic_new = {}
    for letter,number in zip(ls_a,ls_b):
        dic_new[letter] = number
    return dic_new

这将使您拥有所需组合的字典。您通过两个列表，然后可以在其他列表上使用该字典：

List_A = ['A','B','C','D','E']
List_B = [3,2,1,1,2]

dic_new = convert_list(ls_a, ls_b)

other_list = ['a','b','c','d']

for _ in other_list:
    print(dic_new[_.upper()])

# prints
3
2
1
1

欢呼

Answer 4

您可以使用机器学习scikit-learn模块中的解决方案。

OneHotEncoder

LabelEncoder

http://scikit-learn.org/stable/modules/generated/sklearn.preprocessing.OneHotEncoder.html http://scikit-learn.org/stable/modules/generated/sklearn.preprocessing.LabelEncoder.html

熊猫的“硬”方式：

https://stackoverflow.com/a/29330853/9799449

我想在Python中将分类变量转换为数字

4 个答案: