2017-06-19 159 views
2

我正在用熊猫数据框打开,我想将其转换为橙色数据表来补偿丢失的值。 我的数据框看起来像将熊猫数据框转换为橙色数据表

locationId rank Rating type value 
1   1 10  shop 2.668 
2   4 8  store 3.921 
3   3 NAN shop 3.122 

其中排名为序1和5.Type是分类与类型的车间,商店等与评级之间的重复值的整数type.value是浮点。 我想将其转换为橙色数据表,并推算缺少的值。 我也曾看过this但这些功能给我一个错误,并没有为我工作。

回答

1
import numpy as np 
import pandas as pd 
import Orange 
import csv 
from io import StringIO 
from collections import OrderedDict 
from Orange.data import Table, Domain, ContinuousVariable, DiscreteVariable 


def pandas_to_orange(df): 
    domain, attributes, metas = construct_domain(df) 
    orange_table = Orange.data.Table.from_numpy(domain = domain, X = df[attributes].values, Y = None, metas = df[metas].values, W = None) 
    return orange_table 

def construct_domain(df): 
    columns = OrderedDict(df.dtypes) 
    attributes = OrderedDict() 
    metas = OrderedDict() 
    for name, dtype in columns.items(): 

     if issubclass(dtype.type, np.number): 
      if len(df[name].unique()) >= 13 or issubclass(dtype.type, np.inexact) or (df[name].max() > len(df[name].unique())): 
       attributes[name] = Orange.data.ContinuousVariable(name) 
      else: 
       df[name] = df[name].astype(str) 
       attributes[name] = Orange.data.DiscreteVariable(name, values = sorted(df[name].unique().tolist())) 
     else: 
      metas[name] = Orange.data.StringVariable(name) 

    domain = Orange.data.Domain(attributes = attributes.values(), metas = metas.values()) 

    return domain, list(attributes.keys()), list(metas.keys()) 

使用 - > pandas_to_orange(数据帧)

相关问题