2017-06-19 199 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(數據幀)

相關問題