使用pandas将分类值转换为二进制值

2024-06-02 05:18:37 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试使用pandas将分类值转换为二进制值。其思想是将每个唯一的分类值视为一个特征(即列),并根据是否将特定对象(即行)分配给该类别而输入1或0。以下是代码:

data = pd.read_csv('somedata.csv')
converted_val = data.T.to_dict().values()
vectorizer = DV( sparse = False )
vec_x = vectorizer.fit_transform( converted_val )
numpy.savetxt('out.csv',vec_x,fmt='%10.0f',delimiter=',')

我的问题是,如何用列名保存转换后的数据?。在上面的代码中,我可以使用numpy.savetxt函数保存数据,但这只是保存数组,列名将丢失。或者,是否有更有效的方法来执行上述操作?。


Tags: csv数据代码numpypandasdata二进制分类
2条回答

似乎您正在使用scikit learn的DictVectorizer将分类值转换为二进制值。在这种情况下,要将结果与新列名一起存储,可以使用来自vec_x的值和来自DV.get_feature_names()的列构造一个新的数据帧。然后,将数据帧存储到磁盘(例如使用to_csv()),而不是numpy数组。

或者,也可以使用pandas直接使用^{}函数进行编码:

import pandas as pd
data = pd.DataFrame({'T': ['A', 'B', 'C', 'D', 'E']})
res = pd.get_dummies(data)
res.to_csv('output.csv')
print res

输出:

   T_A  T_B  T_C  T_D  T_E
0    1    0    0    0    0
1    0    1    0    0    0
2    0    0    1    0    0
3    0    0    0    1    0
4    0    0    0    0    1

你是说“一个热门”编码?

假设您有以下数据集:

import pandas as pd
df = pd.DataFrame([
            ['green', 1, 10.1, 0], 
            ['red', 2, 13.5, 1], 
            ['blue', 3, 15.3, 0]])

df.columns = ['color', 'size', 'prize', 'class label']
df

enter image description here

现在,你有多种选择。。。

A) 乏味的方法

color_mapping = {
           'green': (0,0,1),
           'red': (0,1,0),
           'blue': (1,0,0)}

df['color'] = df['color'].map(color_mapping)
df

enter image description here

import numpy as np
y = df['class label'].values
X = df.iloc[:, :-1].values
X = np.apply_along_axis(func1d= lambda x: np.array(list(x[0]) + list(x[1:])), axis=1, arr=X)

print('Class labels:', y)
print('\nFeatures:\n', X)

屈服:

Class labels: [0 1 0]

Features:
 [[  0.    0.    1.    1.   10.1]
 [  0.    1.    0.    2.   13.5]
 [  1.    0.    0.    3.   15.3]]

B) Scikit学习DictVectorizer

from sklearn.feature_extraction import DictVectorizer
dvec = DictVectorizer(sparse=False)

X = dvec.fit_transform(df.transpose().to_dict().values())
X

屈服:

array([[  0. ,   0. ,   1. ,   0. ,  10.1,   1. ],
       [  1. ,   0. ,   0. ,   1. ,  13.5,   2. ],
       [  0. ,   1. ,   0. ,   0. ,  15.3,   3. ]])

C) 熊猫

pd.get_dummies(df)

enter image description here

相关问题 更多 >